gpt4 book ai didi

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

转载 作者:太空狗 更新时间:2023-10-29 21:28:21 24 4
gpt4 key购买 nike

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

现在我想保存和恢复经过训练的 DNNClassifier 以供进一步使用。如果有人知道该怎么做,请使用上面链接中的 iris 示例代码告诉我。提前感谢您的帮助!

最佳答案

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

希望对您有所帮助。

关于python - 如何保存和恢复在 TensorFlow python 中训练的 DNNClassifier;虹膜示例,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38358419/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com