Zaktualizuj 'train-mlflow.py'

2022-05-15 14:33:03 +02:00 · 2022-05-15 14:33:03 +02:00 · 3140bbfbf0
commit 3140bbfbf0
parent fbdc99fd9d
1 changed files with 1 additions and 1 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -100,7 +100,7 @@ def my_main(epochs):
            #result.write(f'{y_predicted}')

        input_example = X_testing[0]
-        siganture = (X_training, y_predicted_train))
+        siganture = (X_training, y_predicted_train)
        tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme

        mlflow.pytorch.log_model(model, "model", registered_model_name="s478815", signature=siganture, input_example=input_example)