数据挖掘决策树java

《数据挖掘决策树java》由会员上传分享，免费在线阅读，更多相关内容在教育资源-天天文库。

1、importjava.util.HashMap;importjava.util.HashSet;importjava.util.LinkedHashSet;importjava.util.Iterator;//选自csdn博客//决策树的树结点类classTreeNode{Stringelement;//该值为数据的属性名称Stringvalue;//上一个分裂属性在此结点的值LinkedHashSetchilds;//结点的子结点，以有顺序的链式哈希集存储publicTreeNode(){this.element=null

2、;this.value=null;this.childs=null;}publicTreeNode(Stringvalue){this.element=null;this.value=value;this.childs=null;}publicStringgetElement(){returnthis.element;}publicvoidsetElement(Stringe){this.element=e;}publicStringgetValue(){returnthis.value;}publicvoidsetValue(Stringv)

3、{this.value=v;}publicLinkedHashSetgetChilds(){returnthis.childs;}publicvoidsetChilds(LinkedHashSetchilds){this.childs=childs;}}//决策树类classDecisionTree{TreeNoderoot;//决策树的树根结点publicDecisionTree(){root=newTreeNode();}publicDecisionTree(TreeNoderoot){this.ro

4、ot=root;}publicTreeNodegetRoot(){returnroot;}publicvoidsetRoot(TreeNoderoot){this.root=root;}publicStringselectAtrribute(TreeNodenode,String[][]deData,booleanflags[],LinkedHashSetatrributes,HashMapattrIndexMap){//Gain数组存放当前结点未分类属性的Gain值doubleGain[]=ne

5、wdouble[atrributes.size()];//每条数据中归类的下标，为每条数据的最后一个值intclass_index=deData[0].length-1;//属性名，该结点在该属性上进行分类Stringreturn_atrribute=null;//计算每个未分类属性的Gain值intcount=0;//计算到第几个属性for(Stringatrribute:atrributes){//该属性有多少个值，该属性有多少个分类intvalues_count,class_count;//属性值对应的下标intindex=attrInd

6、exMap.get(atrribute);//存放属性的各个值和分类值LinkedHashSetvalues=newLinkedHashSet();LinkedHashSetclasses=newLinkedHashSet();for(inti=0;i

7、t=values.size();class_count=classes.size();intvalues_vector[]=newint[values_count*class_count];intclass_vector[]=newint[class_count];for(inti=0;i

8、ngc:classes){if(deData[i][class_index].equals(c)){break;}else{k++;}}values_