XGBoostHandler
self, model, prototype_data) XGBoostHandler(
Handler class for creating VetiverModels with xgboost.
Parameters
model : xgboost.Booster
-
a trained and fit xgboost model
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
input_data :
-
Test data
Returns
: prediction
-
Prediction from model