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)