Zaktualizuj 'train-mlflow.py'

2022-05-15 22:41:28 +02:00 · 2022-05-15 22:41:28 +02:00 · a21d7d094e
commit a21d7d094e
parent b3584a5545
1 changed files with 1 additions and 1 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -106,7 +106,7 @@ def my_main(epochs):

        

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