-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
32 lines (28 loc) · 1.03 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import argparse
from core.apriori import Apriori
from core.csv_database import CsvDatabase
from core.factory import Factory
class Setting:
pass
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--input', '-i',
help = 'input file (csv file)')
parser.add_argument('--minsup', '-m', type=float, help='minimum support')
parser.add_argument('--minconf', '-c', type=float, help='minimun confidence')
parser.add_argument('--numrule', '-n', type=int,
help='max number of rules output')
args = parser.parse_args()
db = CsvDatabase(args.input)
Factory.setup_db(db)
apriori = Apriori(db, args.minsup, args.minconf, args.numrule)
frequent_itemsets = apriori.generate_frequent_itemset()
for level in frequent_itemsets:
print(level, ': ', len(frequent_itemsets[level]))
for itemset in frequent_itemsets[level]:
print(itemset)
print('Rules:')
for rule in apriori.generate_all_confidence_rules():
print(rule, '(confidence: %.4f)' % (rule.confidence))
if __name__ == "__main__":
main()