database = TransactionDatabase.loadFromFile("./data/train_adt.csv", ['97'],
100)
data = TransactionDatabase.loadFromFile("./data/train_adt.csv", ['97'], 1)
data1 = TransactionDatabase.loadFromFile("./data/test_adt.csv", ['97'], 1)
# database.cleanAndPrune(2)
# print ("Cleaned database:")
# for transaction in database.transactions:
# print(str(transaction.label))
# print ("\nItems in FP tree and corresponding nodes:")
tree = FPTree()
for t in database:
tree.add(t)
# print(str(tree))
miner = DDPMine(debug=True)
start = time.clock()
Pt = miner.mine(database, 100)
elapsed = time.clock() - start
print("Time Total:%f" % elapsed)
print(Pt)
for row in Pt:
print("Pattern:%s label:%s" % (row[0], row[1]))
for row in Pt:
lb1 = 0
lb2 = 0
for transaction in data.transactions:
if set(row[0]).issubset(set(transaction.itemset)):
if transaction.label == "96":
