File size: 209 Bytes
8c9c141 |
1 2 3 4 5 6 7 8 |
def model_fn(model_dir):
import joblib
import os
return joblib.load(os.path.join(model_dir, "sklearn_model.joblib"))
def predict_fn(input_data, model):
return model.predict(input_data)
|