import numpy as np
from apyori import apriori
if __name__ == '__main__':
data = np.loadtxt("../data/1.txt", dtype=str)
min_supp = 0.1
min_conf = 0.8
min_lift = 0.0
res = apriori(transactions=data, min_support=min_supp, min_confidence=min_conf, min_lift=min_lift)
for rule in res:
print(str(rule))
print("ok")
数据1.txt:
1 2
2 1
2 3
结果:
RelationRecord(items=frozenset({'2'}), support=1.0, ordered_statistics=[OrderedStatistic(items_base=frozenset(), items_add=frozenset({'2'}), confidence=1.0, lift=1.0)])
RelationRecord(items=frozenset({'2', '1'}), support=0.6666666666666666, ordered_statistics=[OrderedStatistic(items_base=frozenset({'1'}), items_add=frozenset({'2'}), confidence=1.0, lift=1.0)])
RelationRecord(items=frozenset({'3', '2'}), support=0.3333333333333333, ordered_statistics=[OrderedStatistic(items_base=frozenset({'3'}), items_add=frozenset({'2'}), confidence=1.0, lift=1.0)])
ok