ium_434704/Jenkinsfile_train

pipeline {
    agent {
        dockerfile true
    } 
    parameters {
        gitParameter branchFilter: 'origin/(.*)', defaultValue: 'master', name: 'BRANCH', type: 'PT_BRANCH'
        string(name: 'epochs', description: 'Model epochs')
        string(name: 'verbose', description: 'Model verbose')
        buildSelector(defaultSelector: lastSuccessful(), description: 'Use latest build', name: 'BUILD_SELECTOR')
    }
    stages {
        stage("Copy artifacts"){
            steps {
                copyArtifacts fingerprintArtifacts: true, projectName: 's434704-create-dataset', selector: buildParameter('BUILD_SELECTOR')
            }
        }
        stage("Run training"){
            steps {
                sh "python3 training.py ${params.verbose}" ${params.epochs}"
            }
        }
        stage('Save trained model files') {
			steps{
				archiveArtifacts 'linear_regression.h5'
			}
		}
    }
    post {
        always {
            mail body: "TRAINING - ${currentBuild.currentResult}", subject: 's434704', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
        }
        success {
            build job: "s434704-evaluation/${params.BRANCH}", parameters: [
                    gitParameter(name: 'BRANCH', value: params.BRANCH)
                    ], wait: false
        }
    }
}