From 1cd58567a038e19f2b4f677ea26ea26f55b10a94 Mon Sep 17 00:00:00 2001 From: Filip Izydorczyk Date: Fri, 4 Jun 2021 16:08:46 +0200 Subject: [PATCH] mlflow signature update --- learning/ml-mlflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/learning/ml-mlflow.py b/learning/ml-mlflow.py index 116a085..70e8298 100644 --- a/learning/ml-mlflow.py +++ b/learning/ml-mlflow.py @@ -75,7 +75,7 @@ with mlflow.start_run(): predicted = model(Variable(torch.from_numpy(x_train))).data.numpy() signature = mlflow.models.signature.infer_signature( - x_train.values, predicted) + x_train, predicted) mlflow.set_experiment("s434700") tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme