mlflow signature update

2021-06-04 16:08:46 +02:00 · 2021-06-04 16:08:46 +02:00 · 1cd58567a0
commit 1cd58567a0
parent 9ac0678cb0
1 changed files with 1 additions and 1 deletions
--- 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