在线预测时允许tf.py_func吗？

Question

在线预测时是否允许tf.py_func？ 如果是，如何使用它的任何例子？ 如果我需要安装额外的 pip 包，答案会改变吗？

我的用例：我处理文本，我需要做词干提取（使用 porter stemmer），我知道如何使用 python 来做，tensorflow 没有 Ops。我想在训练和预测时使用相同的文本处理 - 因此我想将它全部编码成张量流图。

https://www.tensorflow.org/api_docs/python/tf/py_func 有已知的局限性，在我投入更多时间之前，我想知道它在训练和在线预测期间是否有效。

谢谢

Answer 1

很遗憾，没有。 Py_func 无法从已保存的模型中恢复。但是，由于您的用例涉及预处理，因此只需在所有三个（训练、评估、服务）输入函数中显式调用 py_func。如果 py_func 位于图表中间，这将不起作用，但对于词干提取，它应该可以正常工作。