# 构建一个包含10、20、10个单元的三层DNN。
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的文档,在这里。