vetiver.TorchHandler#

class vetiver.TorchHandler(model, ptype_data)#

Handler class for creating VetiverModels with torch.

Parameters

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

describe()

Create description for torch model

handler_predict(input_data, check_ptype)

Generates method for /predict endpoint in VetiverAPI

handler_startup()

Include required packages for prediction

model_class()