ium_434765/mlflow_partner.py
Karolina Oparczyk 933d80f25e
All checks were successful
s434765-training/pipeline/head This commit looks good
s434765-evaluation/pipeline/head This commit looks good
mlflow save model
2021-05-23 22:52:58 +02:00

4 lines
113 B
Python

from mlflow import keras
model = keras.load_model('saved_model.pb')
predict = model.predict([13, 1, 1500, 1500])