vetiver.SKLearnHandler
vetiver.SKLearnHandler#
- class vetiver.SKLearnHandler(model, prototype_data)#
Handler class for creating VetiverModels with sklearn.
- Parameters
model (sklearn.base.BaseEstimator) – a trained sklearn model
- __init__(model, prototype_data)#
Methods
__init__
(model, prototype_data)construct_prototype
()Create data prototype for a model
create_meta
(metadata)Create metadata for a model
describe
()Create description for model
handler_predict
(input_data, check_prototype)Generates method for /predict endpoint in VetiverAPI
handler_startup
()Include required packages for prediction
model_class
()Attributes
pip_name