如何使用contrib.learn.DNNClassifier等新的contrib.learn函数保存模型输出?
skflow.TensorFlowDNNClassifier等不推荐使用的函数有方法.save和.restore。这些应该被迁移到contrib.learn函数,但是我找不到保存和恢复方法。
如果你专门创建变量,你可以使用tf.train.Saver,但如果只使用contrib.learn.DNNClassifier或contrib.learn.DNNRegressor函数,有没有办法保存图形,权重和偏差?
答案 0 :(得分:0)
变量从fit()
保存,并在fit()
,evaluate()
和predict()
的开头恢复。
您是否有需要在这3次调用的上下文之外保存或恢复变量的用例?
答案 1 :(得分:0)
您致电DNNClassifier(..., model_dir)
,然后拨打fit()
和evaluate()
方法。
您拨打DNNClassifier(..., model_dir)
然后可以调用predict()
方法。 您的模型会在model_dir
中找到经过培训的模型,并会加载经过训练的模型参数。