我是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,这将从第一次创建的文件中还原模型.
希望这对你有所帮助.
| 归档时间: |
|
| 查看次数: |
3106 次 |
| 最近记录: |