def setUp(self): super(PrCurveTest, self).setUp() self.logdir = self.get_temp_dir() tf.reset_default_graph() # Generate data. pr_curve_demo.run_all(logdir=self.logdir, steps=3, thresholds=5, verbose=False) # Create a multiplexer for reading the data we just wrote. self.multiplexer = event_multiplexer.EventMultiplexer() self.multiplexer.AddRunsFromDirectory(self.logdir) self.multiplexer.Reload()
def setUp(self): super(PrCurvesPluginTest, self).setUp() logdir = os.path.join(self.get_temp_dir(), "logdir") # Generate data. pr_curve_demo.run_all( logdir=logdir, steps=3, thresholds=5, verbose=False ) # Create a multiplexer for reading the data we just wrote. multiplexer = event_multiplexer.EventMultiplexer() multiplexer.AddRunsFromDirectory(logdir) multiplexer.Reload() context = base_plugin.TBContext(logdir=logdir, multiplexer=multiplexer) self.plugin = pr_curves_plugin.PrCurvesPlugin(context)
def setUp(self): super(PrCurvesPluginTest, self).setUp() logdir = os.path.join(self.get_temp_dir(), 'logdir') # Generate data. pr_curve_demo.run_all( logdir=logdir, steps=3, thresholds=5, verbose=False) # Create a multiplexer for reading the data we just wrote. multiplexer = event_multiplexer.EventMultiplexer() multiplexer.AddRunsFromDirectory(logdir) multiplexer.Reload() context = base_plugin.TBContext(logdir=logdir, multiplexer=multiplexer) self.plugin = pr_curves_plugin.PrCurvesPlugin(context)
def generateDemoData(self): """Generates test data using the plugin demo.""" pr_curve_demo.run_all(logdir=self.logdir, steps=3, thresholds=5, verbose=False)