mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-19 17:17:04 -05:00
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import os
|
|
import pickle
|
|
import json
|
|
import numpy as np
|
|
from sklearn.externals import joblib
|
|
from sklearn.linear_model import Ridge
|
|
|
|
from inference_schema.schema_decorators import input_schema, output_schema
|
|
from inference_schema.parameter_types.numpy_parameter_type import NumpyParameterType
|
|
|
|
|
|
def init():
|
|
global model
|
|
# AZUREML_MODEL_DIR is an environment variable created during deployment.
|
|
# It is the path to the model folder (./azureml-models/$MODEL_NAME/$VERSION)
|
|
# For multiple models, it points to the folder containing all deployed models (./azureml-models)
|
|
model_path = os.path.join(os.getenv('AZUREML_MODEL_DIR'), 'sklearn_regression_model.pkl')
|
|
# deserialize the model file back into a sklearn model
|
|
model = joblib.load(model_path)
|
|
|
|
|
|
input_sample = np.array([[10, 9, 8, 7, 6, 5, 4, 3, 2, 1]])
|
|
output_sample = np.array([3726.995])
|
|
|
|
|
|
@input_schema('data', NumpyParameterType(input_sample))
|
|
@output_schema(NumpyParameterType(output_sample))
|
|
def run(data):
|
|
try:
|
|
result = model.predict(data)
|
|
# you can return any datatype as long as it is JSON-serializable
|
|
return result.tolist()
|
|
except Exception as e:
|
|
error = str(e)
|
|
return error
|