Make the fp_growth module executable in a useful default state.

Eric Naeseth · Eric Naeseth · commit 27554449dbbe · 2009-03-10T15:51:45.000-05:00
diff --git a/fp_growth.py b/fp_growth.py
@@ -357,3 +357,23 @@ def __repr__(self):
             return "<%s (root)>" % type(self).__name__
         return "<%s %r (%r)>" % (type(self).__name__, self.item, self.count)
 
+
+if __name__ == '__main__':
+    from optparse import OptionParser
+    import csv
+    
+    p = OptionParser(usage='%prog data_file')
+    p.add_option('-s', '--minimum-support', dest='minsup', type='int',
+        help='Minimum itemset support (default: 2)')
+    p.set_defaults(minsup=2)
+    
+    options, args = p.parse_args()
+    if len(args) < 1:
+        p.error('must provide the path to a CSV file to read')
+    
+    f = open(args[0])
+    try:
+        for itemset in find_frequent_itemsets(csv.reader(f), options.minsup):
+            print '{' + ', '.join(itemset) + '}'
+    finally:
+        f.close()
