【发布时间】:2019-01-26 03:07:10
【问题描述】:
我在玩Wide and Deep learning example from Tensorflow。我想保存训练有素的分类器以供以后用于预测任务,但我真的不知道怎么做。 DNNLinearCombinedClassifier 没有可用的保存方法,并且酸洗对象也会失败。
任何想法如何保存它?
【问题讨论】:
标签: python tensorflow
我在玩Wide and Deep learning example from Tensorflow。我想保存训练有素的分类器以供以后用于预测任务,但我真的不知道怎么做。 DNNLinearCombinedClassifier 没有可用的保存方法,并且酸洗对象也会失败。
任何想法如何保存它?
【问题讨论】:
标签: python tensorflow
this doc 的检查点保存部分应该会回答您的问题。
【讨论】:
你在找https://www.tensorflow.org/versions/r0.10/how_tos/variables/index.html#saving-and-restoring吗?
特别是保存变量和恢复变量。此外,您可以使用 Checkpoint Files 定期保存权重(如果您的计算机在中途停止训练时很有用,您不必从头开始)。
【讨论】:
model_dir 标志?根据github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/…,model_dir 标志将为您提供:model_dir: Directory to save model parameters, graph and etc. This can also be used to load checkpoints from the directory into a estimator to continue training a previously saved model.,所以我敢打赌,它用于您在深度和广泛的东西中使用的其他分类器。如果是这种情况,那么您可以编写自己的 .py 文件来恢复它并执行您需要做的事情。