machine-learning - model_dir 在 TensorFlow 中指什么？

标签 machine-learning tensorflow

# Build 3 layer DNN with 10, 20, 10 units respectively.
classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,
                                              hidden_units=[10, 20, 10],
                                              n_classes=3,
                                              model_dir="/tmp/iris_model")

model_dir指的是什么？

运行程序后，我找不到/tmp/iris_model。

此片段摘自:https://www.tensorflow.org/get_started/tflearn

最佳答案

model_dir 参数表示保存模型参数、图形等的目录。这也可用于将检查点从目录加载到估计器中，以继续训练之前保存的模型。

就您而言，您的模型保存在系统临时目录中，可能会被系统删除/清理，这就是您找不到它的原因。

我建议您阅读 DNNClassifier 文档 here .

关于machine-learning - model_dir 在 TensorFlow 中指什么？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/44128187/