From b3584a55456aedc264c186f87f7413297343ccf8 Mon Sep 17 00:00:00 2001 From: Kornelia Girejko Date: Sun, 15 May 2022 22:40:15 +0200 Subject: [PATCH] Zaktualizuj 'train-mlflow.py' --- train-mlflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train-mlflow.py b/train-mlflow.py index 5333775..8a59205 100644 --- a/train-mlflow.py +++ b/train-mlflow.py @@ -107,7 +107,7 @@ def my_main(epochs): input_example = X_testing[0] - siganture = mlflow.models.signature.infer_signature(X_training.eval(), y_predicted_train.eval()) + 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 if tracking_url_type_store != "file":