diff --git a/lab09_02_predict.py b/lab09_02_predict.py deleted file mode 100644 index 8443c1d..0000000 --- a/lab09_02_predict.py +++ /dev/null @@ -1,21 +0,0 @@ -import json -import mlflow -import pandas as pd - -import mlflow.pyfunc - -model_name = "country_vaccination" -stage = '1' - -model = mlflow.pyfunc.load_model( - model_uri=f"models:/{model_name}/{stage}" -) - - -with open('country_vaccination/input_example.json', 'r') as datafile: - data = json.load(datafile) - example_input = data["inputs"] - -input_dictionary = {i : x for i, x in enumerate(example_input) } -input_ex = pd.DataFrame(input_dictionary, index=[0]) -print(model.predict(input_ex))