vetiver.TorchHandler#

class vetiver.TorchHandler(model, prototype_data)#

Handler class for creating VetiverModels with torch.

Parameters

model (nn.Module) – a trained torch 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()