model_dir在TensorFlow中引用了什么?

时间:2017-05-23 07:19:12

标签: 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

1 个答案:

答案 0 :(得分:4)

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

在您的情况下,您的模型将保存在系统临时目录中,并且可能会被系统删除/清除,这就是您无法找到它的原因。

我建议您阅读DNNClassifier doc here