Unverified Commit 26ee1a2f authored by John Arevalo's avatar John Arevalo Committed by GitHub
Browse files

Fix #5814

Comments wrongly say dropout is used
parent 5324fc66
...@@ -27,9 +27,8 @@ parser.add_argument('--train_steps', default=1000, type=int, ...@@ -27,9 +27,8 @@ parser.add_argument('--train_steps', default=1000, type=int,
help='number of training steps') help='number of training steps')
def my_model(features, labels, mode, params): def my_model(features, labels, mode, params):
"""DNN with three hidden layers, and dropout of 0.1 probability.""" """DNN with three hidden layers and learning_rate=0.1."""
# Create three fully connected layers each layer having a dropout # Create three fully connected layers.
# probability of 0.1.
net = tf.feature_column.input_layer(features, params['feature_columns']) net = tf.feature_column.input_layer(features, params['feature_columns'])
for units in params['hidden_units']: for units in params['hidden_units']:
net = tf.layers.dense(net, units=units, activation=tf.nn.relu) net = tf.layers.dense(net, units=units, activation=tf.nn.relu)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment