SKLearnHandler

SKLearnHandler(self, model, prototype_data)

Handler class for creating VetiverModels with sklearn.

Parameters

model : sklearn.base.BaseEstimator

a trained sklearn model

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

input_data :

Test data

Returns

prediction :

Prediction from model