classModelHandler(BaseHandler):def__init__(self): self._context =None self.initialized =False# 用於load trained model 的地方,以及其他設定檔definitialize(self,context): self._context = context self.initialized =True# load the model, refer 'custom handler class' above for details# 用於predictions的時候,接收http post request的function: 預設接收key為data的資料defpreprocess(self,data):# Take the input data and make it inference ready preprocessed_data = data[0].get("data")if preprocessed_data isNone: preprocessed_data = data[0].get("body")return preprocessed_data# 用來進行運算的function,最後的結果會交由postprocess()後處理definference(self,model_input):# Do some inference call to engine here and return output model_output = self.model.forward(model_input)return model_output# 指的是發送post:inference()之後的結果defpostprocess(self,inference_output): postprocess_output = inference_outputreturn postprocess_outputdefhandle(self,data,context): model_input = self.preprocess(data) model_output = self.inference(model_input)return self.postprocess(model_output)