s434804/ium_434804
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix 3
Some checks failed
s434804-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Dawid 2021-05-23 18:10:53 +02:00
parent d8d81df090
commit f6465a49f6
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Jenkisfile_train
View File

@ -27,7 +27,7 @@ pipeline {
stage('archiveArtifacts') {
steps{
archiveArtifacts 'vaccines_model/saved_model.pb'
archiveArtifacts 'country_vaccination/**'
archiveArtifacts 'country_vaccination2/**'
}
}
stage('sendMail') {

2
mlflow_model.py
View File

@ -57,8 +57,10 @@ if __name__ == "__main__":
mlf.set_experiment("s434804")
tracking_url_type_store = urlparse(mlf.get_tracking_uri()).scheme
if tracking_url_type_store != "file":
print("no file")
mlf.keras.log_model(model, "country_vaccination2", registered_model_name="s434804", signature=signature,
input_example=input_example)
else:
print("file")
mlf.keras.log_model(model, "vaccines_model", signature=signature, input_example=input_example)
mlf.keras.save_model(model, "country_vaccination", signature=signature, input_example=input_example)