def test_relim(self):
ts1 = perftesting.get_default_transactions()
relim_input = itemmining.get_relim_input(ts1)
report = itemmining.relim(relim_input, 2)
self.assertEqual(17, len(report))
self.assertEqual(6, report[frozenset(['b', 'd'])])
ts2 = perftesting.get_default_transactions_alt()
relim_input = itemmining.get_relim_input(ts2)
report = itemmining.relim(relim_input, 2)
self.assertEqual(19, len(report))
self.assertEqual(5, report[frozenset(['a', 'b'])])
def test_fpgrowth_pruning_off(self):
ts1 = perftesting.get_default_transactions()
fp_input = itemmining.get_fptree(ts1)
report = itemmining.fpgrowth(fp_input, 2, pruning=False)
self.assertEqual(17, len(report))
self.assertEqual(6, report[frozenset(['b', 'd'])])
ts2 = perftesting.get_default_transactions_alt()
fp_input = itemmining.get_fptree(ts2)
report = itemmining.fpgrowth(fp_input, 2, pruning=False)
self.assertEqual(19, len(report))
self.assertEqual(5, report[frozenset(['a', 'b'])])
def test_relim(self):
ts1 = perftesting.get_default_transactions()
relim_input = itemmining.get_relim_input(ts1)
report = itemmining.relim(relim_input, 2)
self.assertEqual(17, len(report))
self.assertEqual(6, report[frozenset(['b', 'd'])])
ts2 = perftesting.get_default_transactions_alt()
relim_input = itemmining.get_relim_input(ts2)
report = itemmining.relim(relim_input, 2)
self.assertEqual(19, len(report))
self.assertEqual(5, report[frozenset(['a', 'b'])])
def test_fpgrowth_pruning_off(self):
ts1 = perftesting.get_default_transactions()
fp_input = itemmining.get_fptree(ts1)
report = itemmining.fpgrowth(fp_input, 2, pruning=False)
self.assertEqual(17, len(report))
self.assertEqual(6, report[frozenset(['b', 'd'])])
ts2 = perftesting.get_default_transactions_alt()
fp_input = itemmining.get_fptree(ts2)
report = itemmining.fpgrowth(fp_input, 2, pruning=False)
self.assertEqual(19, len(report))
self.assertEqual(5, report[frozenset(['a', 'b'])])
def testDefaultSupportConf(self):
ts1 = perftesting.get_default_transactions()
relim_input = itemmining.get_relim_input(ts1)
report = itemmining.relim(relim_input, 2)
rules = assocrules.mine_assoc_rules(report, min_support=2)
self.assertEqual(20, len(rules))
a_rule = (frozenset(['b', 'e']), frozenset(['d']), 2, 1.0)
self.assertTrue(a_rule in rules)
ts2 = perftesting.get_default_transactions_alt()
relim_input = itemmining.get_relim_input(ts2)
report = itemmining.relim(relim_input, 2)
rules = assocrules.mine_assoc_rules(report, min_support=2)
self.assertEqual(20, len(rules))
a_rule = (frozenset(['e']), frozenset(['a', 'd']), 2, 2.0/3.0)
self.assertTrue(a_rule in rules)
def testDefaultSupportConf(self):
ts1 = perftesting.get_default_transactions()
relim_input = itemmining.get_relim_input(ts1)
report = itemmining.relim(relim_input, 2)
rules = assocrules.mine_assoc_rules(report, min_support=2)
self.assertEqual(23, len(rules))
a_rule = (frozenset(['b', 'e']), frozenset(['d']), 2, 1.0)
self.assertTrue(a_rule in rules)
ts2 = perftesting.get_default_transactions_alt()
relim_input = itemmining.get_relim_input(ts2)
report = itemmining.relim(relim_input, 2)
rules = assocrules.mine_assoc_rules(report, min_support=2)
self.assertEqual(29, len(rules))
a_rule = (frozenset(['e']), frozenset(['a', 'd']), 2, 2.0/3.0)
self.assertTrue(a_rule in rules)
