变压器功能不适用于Seldon Core

时间:2019-05-22 02:58:04

标签: kubernetes kubeflow

我正在测试Seldon Core,并且在我的服务模型predict中仅使用MyModel(object)函数就可以启动并运行一个示例模型。 现在,我添加了transform_input函数来预处理输入数据,然后再将输入发送给类。我再次部署了该模型,它看起来运行良好,但是我发现transform_input无法正常工作或未被使用。 我已经找到了docexample,但只显示了添加功能transform_input。我想念什么吗?

1 个答案:

答案 0 :(得分:1)

即使您的模型已经提供了transform_input函数,您仍然需要将其添加为推理图中的TRANSFORMER节点。对于您的特定情况,您需要定义如下图:

  graph:
    name: my-input-transformer
    type: TRANSFORMER
    children:
      - name: my-model
        type: MODEL

两个节点(my-modelmy-input-transformer)都指向相同的Docker映像。