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