TorchHandler

TorchHandler(self, model, prototype_data)

Handler class for creating VetiverModels with torch.

Parameters

Name Type Description Default
model nn.Module a trained torch model required

Methods

Name Description
handler_predict Generates method for /predict endpoint in VetiverAPI

handler_predict

TorchHandler.handler_predict(input_data, check_prototype)

Generates method for /predict endpoint in VetiverAPI

The handler_predict function executes at each API call. Use this function for calling predict() and any other tasks that must be executed at each API call.

Parameters

Name Type Description Default
input_data Test data required

Returns

Type Description
prediction Prediction from model