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

Name	Type	Description
prediction		Prediction from model