From f8025ad3fc8dcf5b1a6231aba3d302ea6ef8b2e5 Mon Sep 17 00:00:00 2001 From: s444501 Date: Fri, 13 May 2022 02:11:38 +0200 Subject: [PATCH] mlflow --- biblioteki_ml.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/biblioteki_ml.py b/biblioteki_ml.py index c756eb4..0358f3e 100644 --- a/biblioteki_ml.py +++ b/biblioteki_ml.py @@ -11,8 +11,7 @@ import pandas as pd # MLFlow mlflow.set_tracking_uri("http://172.17.0.1:5000") -restore_experiment('15') -mlflow.set_experiment("s444501") +mlflow.set_experiment("444501") # Parametry z konsoli @@ -118,13 +117,13 @@ def train_main(epochs, run): if tracking_url_type_store != "file": mlflow.pytorch.log_model(model, - 's444501', - registered_model_name='s444501', + '444501', + registered_model_name='444501', signature=signature, input_example=X_test.numpy()) else: mlflow.pytorch.log_model(model, - 's444501', + '444501', signature=signature, input_example=X_test.numpy())