python - 如何在 tensorflow 中保存和加载 DNN 分类器？

Question

如何在 tensorflow 中保存和加载 DNN 分类器？询问给出的默认虹膜分类器程序。 (https://www.tensorflow.org/get_started/estimator)

Answer 1

要保存并重用分类器，您只需使用相同的 model_dir 路径重新加载它即可。

例如，在您想要使用分类器的方法中，您可以使用相同的 model_dir 再次创建分类器。这将从之前的状态重新加载它。

我使用它进行训练，然后重新加载它来测试单个示例。

tf.estimator.DNNClassifier
(feature_columns=feature_columns, 
hidden_units=[10, 20, 10], 
n_classes=3, 
model_dir="/tmp/iris_model")