def gen_rules(uii, support, confidence):
t0 = time()
# generate associated rules
print('generate associated rules....')
apri = Apriori(uii, support, confidence)
(items_id, allRules) = apri.genRules()
print('Time took for generating rules:', time()-t0, 's')
# items_id 频繁一项集
return (items_id, allRules)
from apriori import Apriori
transactions = [
[1, 2, 5],
[2, 4],
[2, 3],
[1, 2, 4],
[1, 3],
[2, 3],
[1, 3],
[1, 2, 3, 5],
[1, 2, 3]
]
support = 0.2
confidence = 0.4
a = Apriori(transactions, support, confidence)
fs = a.getFrequentSubset()
print('frequent sets......')
for i in fs:
print(i)
print('\n')
print('associate rules......')
rules = a.genRules(fs)
for r in rules:
print(r)
