Apriori 算法 java 的实现

最新推荐文章于 2022-12-18 21:08:26 发布

duckjenny

最新推荐文章于 2022-12-18 21:08:26 发布

阅读量816

点赞数

分类专栏：数据挖掘算法文章标签： apriori java

本文链接：https://blog.csdn.net/duckjenny/article/details/8682128

版权

本文介绍了一个用Java实现的Apriori算法，用于挖掘数据中的频繁项集和关联规则。通过示例数据，展示了如何初始化数据，生成候选集，以及计算规则的过程。

摘要由CSDN通过智能技术生成

     今天花了一天的时间写了个Apriori的算法实现，有点结果，但不完善。基本要素都有了，包括生成频繁集和关联规则~
     放上来跟大家分享个~~吧node.java和apriori.java放一个包里，运行apriori就有结果可观察。

apriori.java

package apriori;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;

public class Apriori {
   String[][] data ={
                      {"beaf","chicken","milk"},
                      {"beaf","cheeze"},
                      {"cheeze","boots"},
                      {"beaf","chicken","cheeze"},
                      {"beaf","chicken","cloth","cheeze","milk"},
                      {"chicken","cloth","milk"},
                      {"chicken","milk","cloth"}
                      };

   ArrayList<ArrayList<String>> D = new ArrayList<ArrayList<String>>();    //main data
   ArrayList<ArrayList<Node>> R = new ArrayList<ArrayList<Node>>();       //frequency sets final result, including all layer
   ArrayList<Node> C = new ArrayList<Node>();               //candidate sets
   ArrayList<Node> F = new ArrayList<Node>();               //frequency sets
   HashSet<String> set = new HashSet<String>();
   int layer =1;
   String split = ",";

   // initiate the string data into arraylist
   void init(){
       int num=0;
       for(int i=0; i<data.length; i++){
           ArrayList<String> al = new ArrayList();
           for(int j=0; j<data[i].length; j++){
               set.add(data[i][j]);
               al.add(data[i][j]);
           }
           D.add(al);
       }
       Iterator<String> it1=set.iterator();
       Node node;
       String temp;
       while(it1.hasNext()){
           temp = it1.next();
           num = 0;
           Iterator<ArrayList<String>> it2 = D.iterator();
           ArrayList<String> al = new ArrayList<String>();
           while(it2.hasNext()){
               if(it2.next().contains(temp))num++;
           }
           node = new Node(temp,num);
           C.add(node);
           if(num>1){
               F.add(n