s434695/ium_434695
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix

Browse Source
This commit is contained in:
s434695 2021-05-24 00:42:36 +02:00
parent 5a6d4f832d
commit 84641091ce
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
vgsales-mlflow.py
View File

@ -70,3 +70,4 @@ with mlflow.start_run():
mlflow.log_metric("rmse", rmse)
#mlflow.keras.log_model(model, 'vgsales_model.h5')
mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train)
mlflow.keras.log_model(model, "model",registered_model_name="s434695", signature= mlflow.models.signature.infer_signature(x_train, y_train), input_example= x_train[0])