Skip to content

Commit 402b561

Browse files
authored
Merge pull request #5844 from johnarevalo/patch-1
Fix #5814
2 parents 5324fc6 + 26ee1a2 commit 402b561

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

samples/core/get_started/custom_estimator.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,8 @@
2727
help='number of training steps')
2828

2929
def my_model(features, labels, mode, params):
30-
"""DNN with three hidden layers, and dropout of 0.1 probability."""
31-
# Create three fully connected layers each layer having a dropout
32-
# probability of 0.1.
30+
"""DNN with three hidden layers and learning_rate=0.1."""
31+
# Create three fully connected layers.
3332
net = tf.feature_column.input_layer(features, params['feature_columns'])
3433
for units in params['hidden_units']:
3534
net = tf.layers.dense(net, units=units, activation=tf.nn.relu)

0 commit comments

Comments
 (0)