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)
