From d8d81df0909d61e20085a10c4a4320d9739e260d Mon Sep 17 00:00:00 2001 From: Dawid Date: Sun, 23 May 2021 18:00:57 +0200 Subject: [PATCH] fix tracking filename --- Jenkisfile_train | 2 +- mlflow_model.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Jenkisfile_train b/Jenkisfile_train index 137c372..5c43511 100644 --- a/Jenkisfile_train +++ b/Jenkisfile_train @@ -27,7 +27,7 @@ pipeline { stage('archiveArtifacts') { steps{ archiveArtifacts 'vaccines_model/saved_model.pb' - archiveArtifacts 'country_vaccinations/**' + archiveArtifacts 'country_vaccination/**' } } stage('sendMail') { diff --git a/mlflow_model.py b/mlflow_model.py index 32c165f..ec39dd8 100644 --- a/mlflow_model.py +++ b/mlflow_model.py @@ -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) \ No newline at end of file + mlf.keras.save_model(model, "country_vaccination", signature=signature, input_example=input_example) \ No newline at end of file