| 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) | |
| 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) | |