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