我正在测试Seldon Core,并且在我的服务模型predict
中仅使用MyModel(object)
函数就可以启动并运行一个示例模型。
现在,我添加了transform_input
函数来预处理输入数据,然后再将输入发送给类。我再次部署了该模型,它看起来运行良好,但是我发现transform_input
无法正常工作或未被使用。
我已经找到了doc和example,但只显示了添加功能transform_input
。我想念什么吗?
答案 0 :(得分:1)
即使您的模型已经提供了transform_input
函数,您仍然需要将其添加为推理图中的TRANSFORMER
节点。对于您的特定情况,您需要定义如下图:
graph:
name: my-input-transformer
type: TRANSFORMER
children:
- name: my-model
type: MODEL
两个节点(my-model
和my-input-transformer
)都指向相同的Docker映像。