tensorflow contrib.learn save

时间:2016-07-14 02:12:24

标签: save tensorflow skflow

如何使用contrib.learn.DNNClassifier等新的contrib.learn函数保存模型输出?

skflow.TensorFlowDNNClassifier等不推荐使用的函数有方法.save和.restore。这些应该被迁移到contrib.learn函数,但是我找不到保存和恢复方法。

如果你专门创建变量,你可以使用tf.train.Saver,但如果只使用contrib.learn.DNNClassifier或contrib.learn.DNNRegressor函数,有没有办法保存图形,权重和偏差?

2 个答案:

答案 0 :(得分:0)

变量从fit()保存,并在fit()evaluate()predict()的开头恢复。

您是否有需要在这3次调用的上下文之外保存或恢复变量的用例?

答案 1 :(得分:0)

培训时

您致电DNNClassifier(..., model_dir),然后拨打fit()evaluate()方法。

测试时

您拨打DNNClassifier(..., model_dir)然后可以调用predict()方法。 您的模型会在model_dir中找到经过培训的模型,并会加载经过训练的模型参数。

参考

Issue #3340 of TF