vetiver.SKLearnHandler#

class vetiver.SKLearnHandler(model, ptype_data)#

Handler class for creating VetiverModels with sklearn.

Parameters

model (sklearn.base.BaseEstimator) – a trained sklearn model

__init__(model, ptype_data)#

Methods

__init__(model, ptype_data)

construct_ptype()

Create data prototype for a model

create_meta([version, url, required_pkgs])

Create metadata for sklearn model

describe()

Create description for sklearn model

handler_predict(input_data, check_ptype)

Generates method for /predict endpoint in VetiverAPI

handler_startup()

Include required packages for prediction

model_class()