vetiver.TorchHandler
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
()