如何保存和加载此对象(regressor
)?
from tensorflow.contrib import learn
regressor = learn.TensorFlowEstimator()
我无法使用tensorflow的默认Saver()来保存它。
如何使用此模型进行增量学习?我对其构造函数中的continue_training
参数感到困惑。它说你可以使用新数据一次又一次地调用fit。同时它提供partial_fit()
。请帮帮我理解?
答案 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}} - 一批样本的增量拟合