Slackbot
10/14/2022, 4:01 AMJiang
10/14/2022, 6:50 AMShihgian Lee
10/14/2022, 1:29 PMmodel = xgboost.get("model:latest")
class MyRunnable(bentoml.Runnable):
SUPPORTED_RESOURCES = ("cpu",)
SUPPORTS_CPU_MULTI_THREADING = True
def __init__(self):
self._predictor = Predictor(model=xgboost.load_model(model))
@bentoml.Runnable.method(batchable=True)
def predict(self, input_df):
predictions = self._predictor.predict(input_df)
return self._build_result(predictions)
def _build_result(self, predictions):
prediction_results = []
for _, row in predictions.iterrows():
prediction_results.append(PredictionResult(rate=row.rate, ...))
return prediction_results
Shihgian Lee
10/17/2022, 11:53 PM