XGBoostHandler

XGBoostHandler(self, model, prototype_data)

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