s430705/ium_430705
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add model registry
Some checks failed
s430705-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
michalzareba 2021-05-23 17:15:38 +02:00
parent d5d573f198
commit a1300fe809
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lab08_mfl.py
View File

@ -82,7 +82,7 @@ with mlflow.start_run():
mlflow.log_metric("RMSE", rmse) mlflow.log_metric("RMSE", rmse)
signature = mlflow.models.signature.infer_signature(X_train.values, model.predict(X_train.values)) signature = mlflow.models.signature.infer_signature(X_train.values, model.predict(X_train.values))
mlflow.keras.save_model(model, "movies_imdb", input_example=input_example, signature=signature) # mlflow.keras.save_model(model, "movies_imdb", input_example=input_example, signature=signature)
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
if tracking_url_type_store != "file": if tracking_url_type_store != "file":