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: bool, **kw)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
check_prototype : boolprediction_type :-
Type of prediction to make. One of “predict”, “predict_proba”, or “predict_log_proba”. Default is “predict”.
Returns
prediction :-
Prediction from model