我想使用保存的TFX管道来生成预测，使用保存的TFX管道对象，因此类似于：
model = load_tfx_model("path/to/artifact")
model.predict(new_data)重要的是，在将输入传递给模型进行推理之前，我希望对输入应用预处理管道（类似于sklearn管道）。
BulkInferrer似乎可以生成预测，我也找到了REST example，但我无法确定这两个选项中的任何一个是否实际应用TFX变换步骤（preprocessing_fn）在新数据上。在我所看到的preprocessing_fn的例子中，like this one，响应变量也在变换中被修改了，但是因为它在预测中是不可用的，我怀疑这个函数只用于训练，有人能解释一下吗？
非常感谢！
1条答案
按热度按时间h7wcgrx31#
BulkInferrer TFX组件对未标记的数据进行批量推理，通常部署在Evaluator组件之后，对经过验证的模型进行推理，或者部署在Trainer组件之后，直接对导出的模型进行推理。
使用训练器组件训练模型或使用Evaluator组件评估模型后，您可以将模型和未标记的数据示例（通常由ExampleGen组件生成）传递到BulkInferrer进行批推理。有关详细信息，请参阅BulkInferrer API。