TensorFlow Estimztor带有额外标签的增量培训

时间:2020-05-27 05:45:33

标签: tensorflow tensorflow-estimator

我目前有一个关于使用TensorFlow进行增量训练的问题。 我想从一个已有的模型开始训练,可以说是训练有2个标签的功能。 输入和输出已经嵌入为整数。

当前,我要实现的是: 1.假设我已经有一个经过训练的带有2个标签的大量数据集的模型。 2.创建一个支持1个额外标签(3个标签)的新模型。 3.从训练有2个标签的旧模型中初始化新模型的变量。 4.继续使用新数据集进行训练。

我在TensorFlow文档中遇到了以下两个API: https://www.tensorflow.org/api_docs/python/tf/estimator/VocabInfo https://www.tensorflow.org/api_docs/python/tf/estimator/WarmStartSettings

如果输入为字符串,并且我们为模型提供了新旧“ vocab”的列表,则似乎可以这样做。

但是,就我而言,我的输入/输出已经嵌入为整数,因此不再以字符串形式出现。似乎上述API并不完全适合我的情况。

我的问题是: 1.如果字符串已经转换为整数,是否可以应用上述API? 2.如果无法实现#1,我还有其他方法可以实现自己的目标吗?

感谢您的帮助。

0 个答案:

没有答案
相关问题