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

change model name
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 17:52:53 +02:00
parent 677a535082
commit bf1f1129fa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlflow_model.py
View File

@ -57,7 +57,7 @@ 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_vaccinations", registered_model_name="s434804", signature=signature,
mlf.keras.log_model(model, "country_vaccinations2", registered_model_name="s434804", signature=signature,
input_example=input_example)
else:
mlf.keras.log_model(model, "vaccines_model", signature=signature, input_example=input_example)