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

Add set experiment
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:19:31 +02:00
parent a1300fe809
commit 1092d08dd9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lab08_mfl.py
View File

@ -9,7 +9,7 @@ from tensorflow.keras.layers import Dense, Dropout
from tensorflow.keras.models import Sequential from tensorflow.keras.models import Sequential
from urllib.parse import urlparse from urllib.parse import urlparse
mlflow.set_experiment("s430705")
mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_tracking_uri("http://172.17.0.1:5000")
@ -83,7 +83,7 @@ with mlflow.start_run():
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)
mlflow.set_experiment("s430705")
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":
mlflow.keras.log_model(model, "movies_imdb2", registered_model_name="s430705", signature=signature, mlflow.keras.log_model(model, "movies_imdb2", registered_model_name="s430705", signature=signature,