13

Could anyone please recommend a good frequent itemset package in python? I only need to find frequent itemset, no need of finding the association rules.

Pluviophile
  • 4,203
  • 14
  • 32
  • 56
Edamame
  • 2,785
  • 5
  • 25
  • 34

3 Answers3

11

I also recommend MLXtend library for frequent itemsets.

usage example:

dataset = [['Milk', 'Onion', 'Nutmeg', 'Kidney Beans', 'Eggs', 'Yogurt'],
           ['Dill', 'Onion', 'Nutmeg', 'Kidney Beans', 'Eggs', 'Yogurt'],
           ['Milk', 'Apple', 'Kidney Beans', 'Eggs'],
           ['Milk', 'Unicorn', 'Corn', 'Kidney Beans', 'Yogurt'],
           ['Corn', 'Onion', 'Onion', 'Kidney Beans', 'Ice cream', 'Eggs']]

te = TransactionEncoder()


te_ary = te.fit(dataset).transform(dataset)


df = pd.DataFrame(te_ary, columns=te.columns_)


frequent_itemsets = apriori(df, min_support=0.1, use_colnames=True)


print frequent_itemsets
Community
  • 1
MoAdel
  • 111
  • 1
  • 2
4

Orange3-Associate package provides frequent_itemsets() function based on FP-growth algorithm.

K3---rnc
  • 3,582
  • 1
  • 14
  • 12
3

MLXtend library has been really useful for me. In its docummentation there is an Apriori implementation that outputs the frequent itemset.

Please check the first example available in http://rasbt.github.io/mlxtend/user_guide/frequent_patterns/apriori/.

tbnsilveira
  • 131
  • 3