Zaktualizuj 'train-mlflow.py'

2022-05-15 15:32:18 +02:00 · 2022-05-15 15:32:18 +02:00 · 1dd1a13f68
commit 1dd1a13f68
parent e1b813d136
1 changed files with 1 additions and 1 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -103,7 +103,7 @@ def my_main(epochs):
            #result.write(f'{y_predicted}')

        input_example = X_testing[1].numpy()
-        siganture = mlflow.models.signature.infer_signature(X_training,model(X_training))
+        siganture = infer_signature(X_training,model(X_training))
        tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme

        if tracking_url_type_store != "file":