diff options
| author |  Jianwei Xie <xiejw@google.com> | 2017-04-13 11:59:54 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-13 13:15:42 -0700 |
| commit | 127fb45e9b29e500a4520300d3b47d1f80b53b4f (patch) | |
| tree | 02637d095104720d68695eabac6acea42f92205d | |
| parent | 4df3a25ab889d1e01ebf6e34092de4c0ce5ca93a (diff) | |
Adds a test to ensure ValidationMonitor will not be added when
min_eval_frequencey is set as 0.
Change: 153095072
| -rw-r--r-- | tensorflow/contrib/learn/python/learn/experiment_test.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/contrib/learn/python/learn/experiment_test.py b/tensorflow/contrib/learn/python/learn/experiment_test.py index e0fafef951..4b5f3a195c 100644 --- a/tensorflow/contrib/learn/python/learn/experiment_test.py +++ b/tensorflow/contrib/learn/python/learn/experiment_test.py @@ -565,6 +565,25 @@ class ExperimentTest(test.TestCase): self.assertEqual([noop_hook], est.eval_hooks) self.assertTrue(isinstance(est.monitors[0], monitors.ValidationMonitor)) + def test_train_and_evaluate_with_no_eval_during_training(self): + for est in self._estimators_for_tests(): + eval_metrics = 'eval_metrics' if not isinstance( + est, core_estimator.Estimator) else None + noop_hook = _NoopHook() + ex = experiment.Experiment( + est, + train_input_fn='train_input', + eval_input_fn='eval_input', + eval_metrics=eval_metrics, + eval_hooks=[noop_hook], + train_steps=100, + eval_steps=100, + min_eval_frequency=0) + ex.train_and_evaluate() + self.assertEqual(1, est.fit_count) + self.assertEqual(1, est.eval_count) + self.assertEqual(0, len(est.monitors)) + def test_min_eval_frequency_defaults(self): def dummy_model_fn(features, labels): # pylint: disable=unused-argument pass |