XGBoostHandler

XGBoostHandler(self, model, prototype_data)

Handler class for creating VetiverModels with xgboost.

Parameters

Name Type Description Default
model xgboost.Booster a trained and fit xgboost model required

Methods

Name Description
handler_predict Generates method for /predict endpoint in VetiverAPI

handler_predict

XGBoostHandler.handler_predict(input_data, check_prototype)

Generates method for /predict endpoint in VetiverAPI

The handler_predict function executes at each API call. Use this function for calling predict() and any other tasks that must be executed at each API call.

Parameters

Name Type Description Default
input_data Test data required

Returns

Type Description
prediction Prediction from model