数据挖掘：数据挖掘Apriori算法

那一抹辉煌

代码:

1. %本文件生成了频繁项集
   
2. clear all;
   
3. clc;
   
4. TotalSheet=[];
   
5. dbstop if error;
   
6. 
   
7. 
   
8. brand_name=csvread('./data/data.csv');
   
9. brand_name=brand_name(:,2);
   
10. brand_name=unique(brand_name);
    
11. 
    
12. 
    
13. 
    
14. thconf=0.6;%最小置信度阈值
    
15. load('./data/table');
    
16. X=table;
    
17. [m,n]=size(X);
    
18. D=X;
    
19. th=ceil(0.1*m);
    
20. %%
    
21. %寻找频繁1项集
    
22. 
    
23. C1=sum(D);
    
24. C1=C1';
    
25. C1=[[1:n]' C1];
    
26. 
    
27. L1=C1;
    
28. support=L1(:,2);
    
29. L1(find(support<th),:)=[];
    
30. 
    
31. TotalSheet{1}=L1;
    
32. 
    
33. %%
    
34. %生成频繁2项集
    
35. 
    
36. C2=combntns(L1(:,1),2);
    
37. temp=[];
    
38. for i=1:size(C2)
    
39.     %项目
    
40.     temp{i,1}=C2(i,:);
    
41.    
    
42.     %支持数
    
43.     sum=0;
    
44.    
    
45.     [p,q]=size(D);
    
46.     for j=1:p
    
47.     if all(D(j,C2(i,:)))
    
48.         sum=sum+1;
    
49.     end
    
50.     end
    
51.     temp{i,2}=sum;
    
52.    
    
53. end
    
54. C2=temp;
    
55. 
    
56. L2=C2;
    
57. support=cell2mat(L2(:,2));
    
58. index=find(support<th);
    
59. L2(index,:)=[];
    
60. 
    
61. TotalSheet{2}=L2;
    
62. 
    
63. %%
    
64. for k=3:6
    
65. 
    
66. 
    
67. 
    
68. %生成频繁k(k>=3)项集
    
69. L3=[];
    
70. C3=cell2mat(L2(:,1));
    
71. 
    
72. [mm,nn]=size(C3);
    
73. item_num=mm;
    
74. if item_num<2
    
75.     break;
    
76. end
    
77. comb_index=combntns(1:item_num,2);
    
78. %按排列组合的方式进行联合
    
79. j=1;
    
80. for i=1:size(comb_index,1)
    
81. comb_temp=union(L2{comb_index(i,:),1});
    
82. len=size(comb_temp,1);
    
83. len2=size(L2{1,1},1);
    
84. %判断是否只差一个元素
    
85. if len~=len2+1
    
86.     continue;
    
87. end
    
88. %判断子集是否都存在
    
89. sub_comb=combntns(comb_temp,len-1);
    
90. 
    
91. if size(intersect(sub_comb,C3,'rows'),1)~=size(sub_comb,1)
    
92.     continue;
    
93. end
    
94. L3{j,1}=comb_temp;
    
95. j=j+1;
    
96. end
    
97. C3=[];
    
98. for i=1:size(L3,1)
    
99.     C3(i,:)=L3{i,1};
    
100. end
     
101. 
     
102. temp=[];
     
103. 
     
104. for i=1:size(C3)
     
105.     %项目
     
106.     temp{i,1}=C3(i,:);
     
107.    
     
108.     %支持数
     
109.     sum=0;
     
110.    
     
111.     for j=1:m
     
112.     if all(D(j,C3(i,:)))
     
113.         sum=sum+1;
     
114.     end
     
115.     end
     
116.     temp{i,2}=sum;
     
117.    
     
118. end
     
119. L3=temp;
     
120. if size(L3,1)==0
     
121.     break;
     
122. end
     
123. 
     
124. support=cell2mat(L3(:,2));
     
125. index=find(support<th);
     
126. L3(index,:)=[];
     
127. 
     
128. 
     
129. TotalSheet{k}=L3;
     
130. L2=L3;
     
131. k
     
132. end

复制代码