当训练数据来自tfrecords且推理来自原始数据时,Tensorflow估计器导出器

时间:2019-02-06 17:17:53

标签: tensorflow

背景

0)我正在研究要导出的NLP模型

1)我有tfRecords形式的训练数据

2)我想导出模型并将其托管在flask应用程序中,所以输入的数据是原始文本

3)我将所有预处理(标记化等)作为我的张量流图的一部分处理

问题

1)鉴于我作为tensorflow图的一部分进行了数据加载(创建数据集和预处理)的事实,因此传入的原始文本会中断该过程吗? (特别是在tf.Dataset创建步骤中)

2)仅加载原始文本而不是tf.Dataset数据是否更有意义?

1 个答案:

答案 0 :(得分:0)

NVM,我完全忘记了您想要一个input_function来馈送数据集,而一个serving_input_function来接受原始数据