TorchHandler

TorchHandler(self, model, prototype_data)

Handler class for creating VetiverModels with torch.

Parameters

model : nn.Module

a trained torch model

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

input_data :

Test data

Returns

: prediction

Prediction from model