def test_trainer_bc(self):
"""Test the `test-ann-batch` subcommands."""
with db.session_scope(META_FILE) as (session, metadata):
tester = TestAnn(self.config)
tester.test_with_hierarchy(self.train_dir, self.ann_dir, 0.001)
correct, total = tester.export_hierarchy_results(self.test_result_batch)
print "Correctly classified: {0}/{1} ({2:.2%})\n".\
format(correct, total, float(correct)/total)
def test_trainer_ae(self):
"""Test the `test-ann` subcommands."""
filter_ = self.config.classification.filter.as_dict()
with db.session_scope(META_FILE) as (session, metadata):
tester = TestAnn(self.config)
tester.test(self.ann_file, self.train_file)
tester.export_results(self.test_result, filter_, 0.001)
