experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix typo in the run_local call of Experiment. Added test for run_local.

Browse Source 
Change: 125458571
This commit is contained in:
Illia Polosukhin 2016-06-21 08:18:50 -08:00 committed by TensorFlower Gardener
parent 002e1854f6
commit 1602ac6a91
2 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/contrib/learn/python/learn/experiment.py
View File

@ -121,9 +121,10 @@ class Experiment(object):
Returns:
The result of the `evaluate` call to the `Estimator`.
"""
self._train_monitors = self._train_monitors or []
if self._local_eval_frequency:
self._train_monitors += [monitors.ValidationMonitor(
input_fn=self._eval_input_fn, steps=self._eval_steps,
input_fn=self._eval_input_fn, eval_steps=self._eval_steps,
metrics=self._eval_metrics, every_n_steps=self._local_eval_frequency
)]
self.train()

19
tensorflow/contrib/learn/python/learn/tests/experiment_test.py
View File

@ -27,6 +27,7 @@ class TestEstimator(object):
def __init__(self):
self.eval_count = 0
self.fit_count = 0
self.monitors = []
def evaluate(self, **kwargs):
tf.logging.info('evaluate called with args: %s' % kwargs)
@ -39,6 +40,8 @@ class TestEstimator(object):
def fit(self, **kwargs):
tf.logging.info('fit called with args: %s' % kwargs)
self.fit_count += 1
if 'monitors' in kwargs:
self.monitors = kwargs['monitors']
return [(key, kwargs[key]) for key in sorted(kwargs.keys())]
@ -115,6 +118,22 @@ class ExperimentTest(tf.test.TestCase):
tf.logging.info('eval duration (expected %f): %f', expected, duration)
self.assertTrue(duration > expected - 0.5 and duration < expected + 0.5)
def test_run_local(self):
est = TestEstimator()
ex = tf.contrib.learn.Experiment(est,
train_input_fn='train_input',
eval_input_fn='eval_input',
eval_metrics='eval_metrics',
train_steps=100,
eval_steps=100,
local_eval_frequency=10)
ex.local_run()
self.assertEquals(1, est.fit_count)
self.assertEquals(1, est.eval_count)
self.assertEquals(1, len(est.monitors))
self.assertTrue(isinstance(est.monitors[0],
tf.contrib.learn.monitors.ValidationMonitor))
if __name__ == '__main__':
tf.test.main()