s444421/ium_444421
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update files

Browse Source
This commit is contained in:
Agata 2022-05-17 21:21:31 +02:00
parent 09d5694438
commit 53320fe8a3
2 changed files with 9 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
training.Jenkinsfile
View File

@ -1,10 +1,10 @@
pipeline {
agent {
dockerfile true
docker {
image 'agakul/ium:mlflow'
args '-v /mlruns:/mlruns'
}
}
options {
copyArtifactPermission('s444421-evaluation');
}
parameters {
buildSelector(
defaultSelector: lastSuccessful(),
@ -23,27 +23,12 @@ pipeline {
checkout([$class: 'GitSCM', branches: [[name: '*/training_and_evaluation']], extensions: [], userRemoteConfigs: [[credentialsId: 's444421', url: 'https://git.wmi.amu.edu.pl/s444421/ium_444421.git']]])
}
}
stage('Script') {
stage('Training') {
steps {
copyArtifacts filter: '*', projectName:'s444421-create-dataset', selector: buildParameter('BUILD_SELECTOR')
sh 'ipython ./training_sacred.py with "epochs=$EPOCHS"'
archiveArtifacts artifacts: 'my_runs/1/*, model.pth, my_runs/_sources/*'
sh 'python training_mlflow.py $EPOCHS'
archiveArtifacts artifacts: 'mlruns/**'
}
}
}
post {
success {
emailext body: 'SUCCESS', subject: 's444421-training status', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
build job: 's444421-evaluation/training_and_evaluation'
}
failure {
emailext body: 'FAILURE', subject: 's444421-training status', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
}
aborted {
emailext body: 'ABORTED', subject: 's444421-training status', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
}
changed {
emailext body: 'CHANGED', subject: 's444421-training status', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
}
}
}

4
training_mlflow.py
View File

@ -118,9 +118,9 @@ def my_main(epochs):
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
if tracking_url_type_store != "file":
mlflow.sklearn.log_model(model, "s444421", registered_model_name="s444421", signature=signature)
mlflow.sklearn.log_model(model, "s444421", registered_model_name="s444421", signature=signature, input_example=X_test.numpy()[:5])
else:
mlflow.sklearn.log_model(model, "s444421", signature=signature)
mlflow.sklearn.log_model(model, "s444421", signature=signature, input_example=X_test.numpy()[:5])
# In[ ]: