ml-finance-python

python scripts for finance machine learning

git clone https://9o.is/git/ml-finance-python.git

fp_growth.py

(1119B)

      1 
      2 import numpy as np
      3 from mlfromscratch.unsupervised_learning import FPGrowth
      4 
      5 def main():
      6     # Demo transaction set
      7     # Example:
      8     # https://en.wikibooks.org/wiki/Data_Mining_Algorithms_In_R/Frequent_Pattern_Mining/The_FP-Growth_Algorithm
      9     
     10     transactions = np.array([
     11         ["A", "B", "D", "E"],
     12         ["B", "C", "E"],
     13         ["A", "B", "D", "E"],
     14         ["A", "B", "C", "E"],
     15         ["A", "B", "C", "D", "E"],
     16         ["B", "C", "D"]
     17     ])
     18 
     19     print ("")
     20     print ("+---------------+")
     21     print ("|   FP-Growth   |")
     22     print ("+---------------+")
     23     min_sup = 3
     24     print ("Minimum Support: %s" % min_sup)
     25     print ("")
     26     print ("Transactions:")
     27     for transaction in transactions:
     28         print ("\t%s" % transaction)
     29 
     30     fp_growth = FPGrowth(min_sup=min_sup)
     31 
     32     print ("")
     33     # Get and print the frequent itemsets
     34     frequent_itemsets = fp_growth.find_frequent_itemsets(
     35         transactions, show_tree=True)
     36 
     37     print ("")
     38     print ("Frequent itemsets:")
     39     for itemset in frequent_itemsets:
     40         print ("\t%s" % itemset)
     41     print ("")
     42 
     43 if __name__ == "__main__":
     44     main()