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

fix tracking filename
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:00:57 +02:00
parent aecb3e0297
commit d8d81df090
2 changed files with 3 additions and 3 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_vaccinations/**'
archiveArtifacts 'country_vaccination/**'
}
}
stage('sendMail') {

4
mlflow_model.py
View File

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