Skip to content

Commit

Permalink
iris example
Browse files Browse the repository at this point in the history
  • Loading branch information
dave dai committed Sep 13, 2017
1 parent 912cfbb commit 456ee18
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -3,4 +3,5 @@

*.gz

iris_*.csv
iris_*.csv
tmp/
2 changes: 1 addition & 1 deletion example/iris.py
Original file line number Diff line number Diff line change
Expand Up @@ -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():
Expand Down

0 comments on commit 456ee18

Please sign in to comment.