diff --git a/cli/score.py b/cli/score.py new file mode 100644 index 00000000..cb3fccba --- /dev/null +++ b/cli/score.py @@ -0,0 +1,24 @@ +import pickle +import json +import numpy +from sklearn.externals import joblib +from sklearn.linear_model import Ridge +from azureml.core.model import Model + +def init(): + global model + # note here "sklearn_regression_model.pkl" is the name of the model registered under + # this is a different behavior than before when the code is run locally, even though the code is the same. + model_path = Model.get_model_path('sklearn_regression_model.pkl') + # deserialize the model file back into a sklearn model + model = joblib.load(model_path) + +# note you can pass in multiple rows for scoring +def run(raw_data): + try: + data = json.loads(raw_data)['data'] + data = numpy.array(data) + result = model.predict(data) + except Exception as e: + result = str(e) + return json.dumps({"result": result.tolist()})