diff options
| -rw-r--r-- | tensorflow/examples/tutorials/monitors/iris_monitors.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/examples/tutorials/monitors/iris_monitors.py b/tensorflow/examples/tutorials/monitors/iris_monitors.py index fdbeac3cd7..a9aa0d4ce0 100644 --- a/tensorflow/examples/tutorials/monitors/iris_monitors.py +++ b/tensorflow/examples/tutorials/monitors/iris_monitors.py @@ -45,7 +45,9 @@ validation_monitor = tf.contrib.learn.monitors.ValidationMonitor( early_stopping_metric_minimize=True, early_stopping_rounds=200) -feature_columns = [tf.contrib.layers.real_valued_column("", dimension=5)] +# Specify that all features have real-value data +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], |