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'
