python - Tensorflow:保存和恢复TensorFlowEstimator()的方法

Question

如何保存和加载此对象(回归器)？

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

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

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

Answer 1

根据这个TF tutorial ，以下内容应该有效:

用于保存:

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

用于恢复:

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

<小时/>

对于增量培训:请参阅以下详细信息。他们给出了很好的解释。

continue_training - 将其设置为True，模型初始化一次，并且将在每次调用 fit 时持续进行训练。

partial_fit - 增量拟合一批 sample