s434700/ium_s434700
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

mlflow signature update

Browse Source
This commit is contained in:
Filip Izydorczyk 2021-06-04 16:08:46 +02:00
parent 9ac0678cb0
commit 1cd58567a0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
learning/ml-mlflow.py
View File

@ -75,7 +75,7 @@ with mlflow.start_run():
predicted = model(Variable(torch.from_numpy(x_train))).data.numpy() predicted = model(Variable(torch.from_numpy(x_train))).data.numpy()
signature = mlflow.models.signature.infer_signature( signature = mlflow.models.signature.infer_signature(
x_train.values, predicted) x_train, predicted)
mlflow.set_experiment("s434700") mlflow.set_experiment("s434700")
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme