Zaktualizuj 'train-mlflow.py'

2022-05-15 15:06:33 +02:00 · 2022-05-15 15:06:33 +02:00 · c544e57fe1
commit c544e57fe1
parent 3140bbfbf0
1 changed files with 1 additions and 1 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -99,7 +99,7 @@ def my_main(epochs):
            #result = open("output",'w+')
            #result.write(f'{y_predicted}')

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