保存宽而深的张量流模型

Question

我正在玩Wide and Deep learning example from Tensorflow。我想保存训练好的分类器，以便以后用于预测任务，但我真的不知道如何。 DNNLinearCombinedClassifier没有可用的保存方法，并且对该对象进行pickle也会失败。

任何想法如何保存？

Answer 1

this doc的检查点保存部分应该回答您的问题。

Answer 2

您在寻找https://www.tensorflow.org/versions/r0.10/how_tos/variables/index.html#saving-and-restoring吗？

特别是保存变量和还原变量。此外，您可以使用检查点文件定期保存权重（如果您的计算机中途停止训练，则无用，您不必从头开始）。