update predict.py
Some checks are pending
s444356-training/pipeline/head Build queued...
s444356-evaluation/pipeline/head This commit looks good

This commit is contained in:
Maciej Czajka 2022-05-11 21:19:17 +02:00
parent da3554455e
commit cc720a8dee

View File

@ -3,7 +3,7 @@ import mlflow
import sys import sys
import numpy as np import numpy as np
input = sys.argv[1] #input = sys.argv[1]
logged_model = 'mlruns/1/70439eb482b54d56b54b0ecc6f1ca96f/artifacts/s444409' logged_model = 'mlruns/1/70439eb482b54d56b54b0ecc6f1ca96f/artifacts/s444409'
loaded_model = mlflow.pyfunc.load_model(logged_model) loaded_model = mlflow.pyfunc.load_model(logged_model)