1. 必须包含 execute 函数:def execute(input_data, params=None):
-2. 函数接收 input_data(字典)和 params(字典,可选)参数
-3. 函数返回处理结果(字典)
-4. 示例:
-def execute(input_data, params=None):
- a = input_data.get('a', 0)
- b = input_data.get('b', 0)
- return {"result": a + b}
-
- 1. 必须包含 execute 函数:def execute(input_data, params=None):
-2. 函数接收 input_data(字典)和 params(字典,可选)参数
-3. 函数返回处理结果(字典)
-4. 示例(PyTorch模型):
-import torch
-
-# 加载模型
-model = torch.load('model/model.pth')
-model.eval()
-
-def execute(input_data, params=None):
- data = torch.tensor(input_data['data'])
- with torch.no_grad():
- output = model(data)
- return {"result": output.numpy().tolist()}
-
-