关联规则频繁项集挖掘增加测试数据

Author: Boyce.Zhang

src/apriori/Itemset.java

@@ -112,9 +112,9 @@ public Itemset frequent_gen(Transactions transactions) {
             if (sup >= transactions.minsup()) {
                 frequentItemset.addItem(item);
 
-                System.out.println("> add    a item: " + item + ", sup=" + count + "/" + n);
+                System.out.println("> add    a item: " + item + ", sup=" + count + "/" + n + " = " + (double)count/n);
             } else  {
-                System.out.println("> remove a item: " + item + ", sup=" + count + "/" + n);
+                System.out.println("> remove a item: " + item + ", sup=" + count + "/" + n + " = " + (double)count/n);
             }
         }
         System.out.println();

src/apriori/example/Calculator.java

@@ -0,0 +1,63 @@
+package apriori.example;
+
+import apriori.Itemset;
+import apriori.Transaction;
+import apriori.Transactions;
+import common.utils.FileUtils;
+import common.utils.StringUtils;
+
+import java.io.File;
+import java.util.*;
+
+/**
+ * Created with IntelliJ IDEA.
+ * User: Boyce
+ * Date: 7/12/15
+ * Time: 14:37
+ * To change this template use File | Settings | File Templates.
+ */
+public class Calculator {
+    private final static String BASE_PATH = "/Users/Boyce/GitProjects/Algorithm/src/apriori/example/";
+
+    public static void main(String[] args) {
+        Transactions transactions = new Transactions(0.9, 0.4);
+
+        File file = new File(BASE_PATH + "data/accidents.dat");
+        List<String> lines = FileUtils.readLines(file);
+
+        Transaction t;
+        for (int i=0;i<lines.size(); i++) {
+            String line = lines.get(i);
+            if (StringUtils.isNotEmpty(line)) {
+                t = new Transaction("t" + i, transactions);
+
+                String[] items = line.split(" ");
+                int length = items.length;
+
+                for (int j=0; j<length; j++) {
+                    t.addItem(items[j]);
+                }
+            }
+        }
+
+        System.out.println(transactions);
+        System.out.println(transactions.n());
+
+        // for
+        Itemset c = Itemset.init(transactions);
+        System.out.println(c);
+
+        Itemset f = c.frequent_gen(transactions);
+        System.out.println(f);
+
+        while (!c.isEmpty()) {
+            c = f.candidate_gen();
+            System.out.println("candidate set: " + c);
+
+            if (!c.isEmpty()) {
+                f = c.frequent_gen(transactions);
+                System.out.println("frequent  set: " + f);
+            }
+        }
+    }
+}