33 lines
1.1 KiB
Plaintext
33 lines
1.1 KiB
Plaintext
pipeline {
|
|
agent {
|
|
docker { image 's444507_create_dataset_image:latest' }
|
|
}
|
|
parameters {
|
|
string(name: 'epoch', defaultValue: '1000', description: 'Number of epochs to train model.')
|
|
}
|
|
stages {
|
|
stage('Get arifacts') {
|
|
steps {
|
|
copyArtifacts fingerprintArtifacts: true, projectName: 's444507-create-dataset', selector: lastSuccessful()
|
|
}
|
|
}
|
|
stage('Train model with sacred') {
|
|
steps {
|
|
sh "python3 lab08_deepLearining_mlflow.py $epoch'"
|
|
archiveArtifacts artifacts: 'games_model.pkl'
|
|
archiveArtifacts artifacts: 'mlruns/**'
|
|
archiveArtifacts artifacts: 'my_model/**'
|
|
sh 'rm -r mlruns'
|
|
sh 'rm -r my_model'
|
|
}
|
|
}
|
|
}
|
|
post {
|
|
success {
|
|
archiveArtifacts artifacts: 'CarPrices_pytorch_model.pkl, mlruns/**, my_model/**', followSymlinks: false
|
|
}
|
|
always {
|
|
emailext body: "${currentBuild.currentResult}", subject: 's444507-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
|
|
}
|
|
}
|
|
} |