Tensorflow: 保存和恢复TensorFlowEstimator的方法

如何保存和加载此对象（regressor）？

from tensorflow.contrib import learnregressor = learn.TensorFlowEstimator()

我无法使用TensorFlow的默认Saver()来保存它。

如何使用此模型进行增量学习？我对其构造函数中的continue_training参数感到困惑。它说你可以一次又一次地调用fit来处理新数据。同时，它还提供了partial_fit()。请帮助我理解？

回答：

根据这个TF教程，以下方法应该有效：

保存：

regressor.save('/tmp/tf_examples/my_model_1/')

恢复：

new_regressor = TensorFlowEstimator.restore('/tmp/tf_examples/my_model_2')

对于增量训练：请参考以下详细信息。他们提供了非常好的解释。

continue_training – 将其设置为True，模型将在初始化一次后，每次调用fit时持续训练。

partial_fit – 对一批样本进行增量拟合

学技术