def testSummaryWriting(self):
     est = estimator.Estimator(model_fn=linear_model_fn)
     est.fit(input_fn=boston_input_fn, steps=200)
     est.evaluate(input_fn=boston_input_fn, steps=200)
     loss_summary = util_test.simple_values_from_events(
         util_test.latest_events(est.model_dir), ['OptimizeLoss/loss'])
     self.assertEqual(1, len(loss_summary))
Example #2
0
 def testSummaryWriting(self):
   est = estimator.Estimator(model_fn=linear_model_fn)
   est.fit(input_fn=boston_input_fn, steps=200)
   est.evaluate(input_fn=boston_input_fn, steps=200)
   loss_summary = util_test.simple_values_from_events(
       util_test.latest_events(est.model_dir), ['OptimizeLoss/loss'])
   self.assertEqual(1, len(loss_summary))