From 456ee1835e527dad9d8b0f1dd2d2a3bc18fb41b4 Mon Sep 17 00:00:00 2001 From: dave dai Date: Wed, 13 Sep 2017 17:12:04 +0800 Subject: [PATCH] iris example --- .gitignore | 3 ++- example/iris.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 3db26be..f8c2149 100644 --- a/.gitignore +++ b/.gitignore @@ -3,4 +3,5 @@ *.gz -iris_*.csv \ No newline at end of file +iris_*.csv +tmp/ \ No newline at end of file diff --git a/example/iris.py b/example/iris.py index cead4fc..105bbd7 100644 --- a/example/iris.py +++ b/example/iris.py @@ -42,7 +42,7 @@ def main(): feature_columns = [tf.contrib.layers.real_valued_column("", dimension=4)] # 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") + classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,hidden_units=[10, 20, 10],n_classes=3,model_dir="tmp/iris_model") # Define the training inputs def get_train_inputs():