save - tensorflow contrib.learn 保存

Question

如何使用新的 contrib.learn 函数（如 contrib.learn.DNNClassifier）保存模型输出？

skflow.TensorFlowDNNClassifier 等已弃用的函数具有 .save 和 .restore 方法。据说这些已迁移到 contrib.learn 函数，但不再有我能找到的保存和恢复方法。

如果您专门创建变量，您可以使用 tf.train.Saver，但是如果您只使用 contrib.learn.DNNClassifier 或 contrib.learn.DNNRegressor 函数，有什么方法可以保存您的图表、权重和偏差？

Answer 1

变量从 保存，并在、和fit()的开头恢复。fit()evaluate()predict()

您是否有需要在这 3 个调用的上下文之外保存或恢复变量的用例？

Answer 2

训练时

您调用DNNClassifier(..., model_dir)然后调用fit()andevaluate()方法。

测试时

您调用DNNClassifier(..., model_dir)然后可以调用predict()方法。您的模型将在 中找到经过训练的模型，model_dir并将加载经过训练的模型参数。

参考

TF 的第 3340 期