From a21d7d094e0e919e28d7155f1007eb4835a3898a Mon Sep 17 00:00:00 2001 From: Kornelia Girejko Date: Sun, 15 May 2022 22:41:28 +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 8a59205..1d56809 100644 --- 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