导出具有多个服务功能的 DNNLinearCombinedClassifier

Question

我正在训练一个 DNNLinearCombinedClassifier，并通过 tensorflow-serving 为它提供低延迟预测服务。当前的服务签名需要一个序列化的 tf.example。

我希望能够导出具有多个 serving_fn 的同一模型，因为我想更改预期的数据格式。

我不能使用ServingInputReceiverreceiver_tensors_alternatives，因为我需要根据提供的数据格式对我的功能应用稍微不同的转换。这些转变主要是在重塑我的特征。

有没有办法导出一个提供多个不同ServingInputReceivers 的模型？

Answer 1

IIUC，您希望同一模型有两个不同的签名，并且您希望对每个签名的输入应用不同的转换，对吗？您可以在服务代码中执行此操作，但您要确保在训练期间应用相同的转换，并尝试在两个地方复制代码为训练/服务偏差打开了大门。

一种选择是分支您的 TFX 管道，以便您有两个 Transform 为两个 Trainer 提供数据，并为您的模型添加不同的转换。