class TestExperimentKB(unittest.TestCase):
def setUp(self):
self.data_paths = ['tests/data/n3/data.n3', 'tests/data/n3/ontology/numbers.n3']
self.graph = load.rdf(self.data_paths)
self.kb = ExperimentKB(self.graph, scorefunctions.lift)
def test_examples(self):
self.assertEqual(len(self.kb.examples), 6)
def test_hierarchy_members(self):
root = self.kb.get_root()
self.assertEqual(self.kb.n_examples(), self.kb.n_members(root.label))
class TestExperimentKB(unittest.TestCase):
def setUp(self):
self.data_paths = [
'tests/data/n3/data.n3', 'tests/data/n3/ontology/numbers.n3'
]
self.graph = load.rdf(self.data_paths)
self.kb = ExperimentKB(self.graph, scorefunctions.lift)
def test_examples(self):
self.assertEqual(len(self.kb.examples), 6)
def test_hierarchy_members(self):
root = self.kb.get_root()
self.assertEqual(self.kb.n_examples(), self.kb.n_members(root.label))
def run(kwargs, cli=False):
if cli:
logger.setLevel(logging.DEBUG if kwargs['verbose'] else logging.INFO)
else:
logger.setLevel(logging.NOTSET)
logger.info('Starting Hedwig')
start = time.time()
start_date = datetime.now().isoformat()
graph = build_graph(kwargs)
logger.info('Building the knowledge base')
score_func = getattr(scorefunctions, kwargs['score'])
kb = ExperimentKB(graph, score_func, instances_as_leaves=kwargs['leaves'])
validator = Validate(kb,
significance_test=significance.apply_fisher,
adjustment=getattr(adjustment, kwargs['adjust']))
rules_per_target = run_learner(kwargs, kb, validator)
rules_report = generate_rules_report(kwargs, rules_per_target)
end = time.time()
time_taken = end - start
logger.info('Finished in %d seconds' % time_taken)
logger.info('Outputing results')
if kwargs['covered']:
with open(kwargs['covered'], 'w') as f:
examples = Rule.ruleset_examples_json(rules_per_target)
f.write(json.dumps(examples, indent=2))
parameters_report = _parameters_report(kwargs, start_date, time_taken)
rules_out_file = kwargs['output']
if rules_out_file:
with open(rules_out_file, 'w') as f:
if rules_out_file.endswith('json'):
f.write(
Rule.to_json(rules_per_target, show_uris=kwargs['uris']))
else:
f.write(parameters_report)
f.write(rules_report)
elif cli:
print parameters_report
print rules_report
return rules_per_target
def setUp(self):
self.data_paths = ['tests/data/n3/data.n3', 'tests/data/n3/ontology/numbers.n3']
self.graph = load.rdf(self.data_paths)
self.kb = ExperimentKB(self.graph, scorefunctions.lift)
def setUp(self):
self.ontology_paths = ['tests/data/csv/ontology/Hierarchy.tsv']
self.data_path = 'tests/data/csv/Cities_clusters.csv'
graph = load.csv(self.ontology_paths, self.data_path)
self.kb = ExperimentKB(graph, scorefunctions.lift)
def setUp(self):
self.data_paths = [
'tests/data/n3/data.n3', 'tests/data/n3/ontology/numbers.n3'
]
self.graph = load.rdf(self.data_paths)
self.kb = ExperimentKB(self.graph, scorefunctions.lift)
