Skip to content

GitLab

Commit db0de3bb authored by Neal Wu's avatar Neal Wu
Browse files

Train wide_deep.py for longer for more consistent results that match the tutorials

parent a6e1b895
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/wide_deep/wide_deep.py
View file @ db0de3bb
...@@ -44,7 +44,7 @@ parser.add_argument( ...@@ -44,7 +44,7 @@ parser.add_argument(
help="Valid model types: {'wide', 'deep', 'wide_deep'}.") help="Valid model types: {'wide', 'deep', 'wide_deep'}.")
parser.add_argument( parser.add_argument(
'--train_epochs', type=int, default=20, help='Number of training epochs.') '--train_epochs', type=int, default=40, help='Number of training epochs.')
parser.add_argument( parser.add_argument(
'--epochs_per_eval', type=int, default=2, '--epochs_per_eval', type=int, default=2,
...@@ -213,7 +213,7 @@ def main(unused_argv): ...@@ -213,7 +213,7 @@ def main(unused_argv):
# Display evaluation metrics # Display evaluation metrics
print('Results at epoch', (n + 1) * FLAGS.epochs_per_eval) print('Results at epoch', (n + 1) * FLAGS.epochs_per_eval)
print('-' * 30) print('-' * 60)
for key in sorted(results): for key in sorted(results):
print('%s: %s' % (key, results[key])) print('%s: %s' % (key, results[key]))
......
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