if __name__ == "__main__":
# Init database and field list
db = Database(['A', 'B', 'C', 'D', 'E', 'F'])
# Create transactions
t1 = Transaction(0, frozenset(['A', 'B', 'C']))
t2 = Transaction(1, frozenset(['A', 'C']))
t3 = Transaction(2, frozenset(['A', 'D']))
t4 = Transaction(3, frozenset(['B', 'E', 'F']))
# Add transactions to database
db.add_transaction(t1)
db.add_transaction(t2)
db.add_transaction(t3)
db.add_transaction(t4)
# Output database
db.print_as_set()
db.print_as_boolean()
# Init and run Apriory
alg = AprioriAlgorithm(db, 0.30, 0.50)
alg.run()
# Print results
print('Support of frequent itemsets:')
alg.print_support()
print('Find rules:')
alg.print_rules(top=10)
