Tensorflow:保存和恢复TensorFlowEstimator()

时间:2016-10-09 18:28:05

标签: python python-3.x machine-learning tensorflow lstm

如何保存和加载此对象(regressor)?

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

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

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

1 个答案:

答案 0 :(得分:1)

根据此continue_training,以下内容应该有效:

保存:

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

用于恢复:

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

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

partial_fit - 将此设置为True,模型初始化一次,并且会在每次调用时不断训练。

{{3}} - 一批样本的增量拟合