# 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
。
答案 0 :(得分:4)
model_dir
参数表示保存模型参数,图形等的目录。这也可用于将检查点从目录加载到估算器中,以继续训练先前保存的模型。
在您的情况下,您的模型将保存在系统临时目录中,并且可能会被系统删除/清除,这就是您无法找到它的原因。
我建议您阅读DNNClassifier doc here。