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 |