def testSupport5(self): ts1 = perftesting.get_default_transactions() relim_input = itemmining.get_relim_input(ts1) report = itemmining.relim(relim_input, 5) rules = assocrules.mine_assoc_rules(report, min_support=5) self.assertEqual(2, len(rules)) a_rule = (frozenset(['d']), frozenset(['b']), 6, 0.75) self.assertTrue(a_rule in rules)
def testSupport5(self): ts1 = perftesting.get_default_transactions() relim_input = itemmining.get_relim_input(ts1) report = itemmining.relim(relim_input, 5) rules = assocrules.mine_assoc_rules(report, min_support=5) self.assertEqual(2, len(rules)) a_rule = (frozenset(['d']), frozenset(['b']), 6, 0.75) self.assertTrue(a_rule in rules)
def testConfidence075(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, min_confidence=0.75) self.assertEqual(8, len(rules)) a_rule = (frozenset(["b"]), frozenset(["d"]), 6, 0.75) self.assertTrue(a_rule in rules)
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)
def Apriori_three(data_tuple): transactions = perftesting.get_default_transactions() relim_input = itemmining.get_relim_input(transactions) item_sets = itemmining.relim(relim_input, min_support=50)#////最小关联度 rules = assocrules.mine_assoc_rules(item_sets, min_support=2, min_confidence=0.5) print(rules)
from pymining import itemmining from pymining import seqmining from pymining import assocrules, perftesting import originalApriori import originalFPgrowth transactions = perftesting.get_default_transactions() transactions = (('1', '3', '4'), ('2', '3', '5'), ('1', '2', '3', '5'), ('2', '5')) print 'transactions' for item in transactions: print item relim_input = itemmining.get_relim_input(transactions) print 'relim_input' print relim_input # report=itemmining.relim(relim_input,min_support=2) # print report transactionslist = [] for item in transactions: tmplist = [] for i in item: tmplist.append(i) transactionslist.append(tmplist) # transactionslist=[['a','b','c'],['b'],['a'],['a','c','d'],['b','c'],['b','c']] # seqs=('caabc','abcb','cabc','abbca') print 'transactionslist' transactionslist = [['1', '3', '4'], ['2', '3', '5'], ['1', '2', '3', '5'], ['2', '5']] for item in transactionslist: print item # freq_seqs=seqmining.freq_seq_enum(seqs,min_support=2) # print sorted(freq_seqs)
from pymining import itemmining from pymining import seqmining from pymining import assocrules,perftesting import originalApriori import originalFPgrowth transactions=perftesting.get_default_transactions() transactions=(('1', '3','4'), ('2', '3', '5'), ('1', '2', '3', '5'), ('2', '5')) print 'transactions' for item in transactions: print item relim_input=itemmining.get_relim_input(transactions) print 'relim_input' print relim_input # report=itemmining.relim(relim_input,min_support=2) # print report transactionslist=[] for item in transactions: tmplist=[] for i in item: tmplist.append(i) transactionslist.append(tmplist) # transactionslist=[['a','b','c'],['b'],['a'],['a','c','d'],['b','c'],['b','c']] # seqs=('caabc','abcb','cabc','abbca') print 'transactionslist' transactionslist=[['1', '3','4'], ['2', '3', '5'], ['1', '2', '3', '5'], ['2', '5']] for item in transactionslist: print item # freq_seqs=seqmining.freq_seq_enum(seqs,min_support=2) # print sorted(freq_seqs) print 'assocrules'