ium_434742/predict-s434704.py
patrycjalazna 00fcfd4d1c
Some checks failed
s434742-training/pipeline/head There was a failure building this commit
s434742-evaluation/pipeline/head This commit looks good
added model to registry, prediction on s434704
2021-05-23 23:01:50 +02:00

13 lines
269 B
Python

import json
import sys
import mlflow
filename = sys.argv[1]
model = mlflow.keras.load_model("movies_on_streaming_platforms_model")
path = 'movies_on_streaming_platforms_model/' + filename
with open(path) as f:
data = json.load(f)
model.predict(data['inputs'])