Zaktualizuj 'train-mlflow.py'

2022-05-15 22:22:44 +02:00 · 2022-05-15 22:22:44 +02:00 · 28e50142f0
commit 28e50142f0
parent 3d0621f745
1 changed files with 1 additions and 1 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -112,7 +112,7 @@ def my_main(epochs):
        mlflow.log_metric("rmse", rmse)
        mlflow.log_metric("mae", mae)

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