SKLearnHandler

SKLearnHandler(self, model, prototype_data)

Handler class for creating VetiverModels with sklearn.

Parameters

Name Type Description Default
model sklearn.base.BaseEstimator a trained sklearn model required

Methods

Name Description
handler_predict Generates method for /predict endpoint in VetiverAPI

handler_predict

SKLearnHandler.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 from model