From 3e34456d79f6aa926d2ed9994b6e21a6e3ad2261 Mon Sep 17 00:00:00 2001 From: Dawid Date: Sun, 23 May 2021 18:50:55 +0200 Subject: [PATCH] copy model from mlflow script --- pred_from_mlflow_model.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 pred_from_mlflow_model.py diff --git a/pred_from_mlflow_model.py b/pred_from_mlflow_model.py new file mode 100644 index 0000000..c7f91a8 --- /dev/null +++ b/pred_from_mlflow_model.py @@ -0,0 +1,18 @@ +import mlflow as mlf +import json +import mlflow as mlf +import pandas as pd + +model_name = "movies_imdb" +model_version = 1 + +model = mlf.pyfunc.load_model( + model_uri=f"models:/{model_name}/{model_version}" +) + +with open('movies_imdb/input_example.json', 'r') as datafile: + data = json.load(datafile)['inputs'] + +input_dictionary = {idx:x for idx, x in enumerate(data)} +input_ex = pd.DataFrame([input_dictionary]) +print(model.predict(input_ex))