如何保存和恢复在TensorFlow python中训练的DNNClassifier; 虹膜的例子

Kaz*_*uki 7 python tensorflow

我是TensorFlow的新手,几天前才开始学习.我已完成本教程(https://www.tensorflow.org/versions/r0.9/tutorials/tflearn/index.html#tf-contrib-learn-quickstart)并将完全相同的想法应用于我自己的数据集.(出来很不错!)

现在,我想保存并恢复经过培训的DNNClassifier以供进一步使用.如果有人知道如何操作,请使用上面链接中的iris示例代码告诉我.感谢您的帮助!

Lui*_*eal 11

找到了解决方案吗?如果没有,您可以在创建DNNClassifier时在构造函数上指定model_dir参数,这将创建此目录中的所有检查点和文件(保存步骤).如果要执行还原步骤,只需创建另一个传递相同model_dir参数(还原阶段)的DNNClassifier,这将从第一次创建的文件中还原模型.

希望这对你有所帮助.