diff --git a/Jenkisfile_train b/Jenkisfile_train index 5c43511..879942b 100644 --- a/Jenkisfile_train +++ b/Jenkisfile_train @@ -27,7 +27,7 @@ pipeline { stage('archiveArtifacts') { steps{ archiveArtifacts 'vaccines_model/saved_model.pb' - archiveArtifacts 'country_vaccination/**' + archiveArtifacts 'country_vaccination2/**' } } stage('sendMail') { diff --git a/mlflow_model.py b/mlflow_model.py index ec39dd8..e989f76 100644 --- a/mlflow_model.py +++ b/mlflow_model.py @@ -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) \ No newline at end of file