决策树算法 java源码.doc

《决策树算法 java源码.doc》由会员上传分享，免费在线阅读，更多相关内容在教育资源-天天文库。

1、importjava.util.HashMap;importjava.util.HashSet;importjava.util.LinkedHashSet;importjava.util.Iterator; //1、selectAtrribute中的一个数组下标出错2、两个字符串相等的判断//3、输入的数据有一个错误4、selectAtrribute中最后一个循环忘记了i++ //决策树的树结点类classTreeNode{Stringelement; //该值为数据的属性名称Stringvalue;   //上一个分裂属性在此结点的值LinkedHashS

2、etchilds; //结点的子结点，以有顺序的链式哈希集存储 publicTreeNode(){this.element=null;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;

3、}publicStringgetValue(){returnthis.value;}publicvoidsetValue(Stringv){this.value=v;}publicLinkedHashSetgetChilds(){returnthis.childs;}publicvoidsetChilds(LinkedHashSetchilds){this.childs=childs;}} //决策树类classDecisionTree{TreeNoderoot; //决策树的树根结点publicDecisionTre

4、e(){root=newTreeNode();}publicDecisionTree(TreeNoderoot){this.root=root;}publicTreeNodegetRoot(){returnroot;}publicvoidsetRoot(TreeNoderoot){this.root=root;}publicStringselectAtrribute(TreeNodenode,String[][]deData,booleanflags[],LinkedHashSetatrributes,HashMap

5、er>attrIndexMap){//Gain数组存放当前结点未分类属性的Gain值doubleGain[]=newdouble[atrributes.size()];//每条数据中归类的下标，为每条数据的最后一个值intclass_index=deData[0].length-1;//属性名，该结点在该属性上进行分类Stringreturn_atrribute=null;//计算每个未分类属性的Gain值intcount=0;//计算到第几个属性for(Stringatrribute:atrributes){//该属性有多少个值，该属性有多少个分类intv

6、alues_count,class_count;//属性值对应的下标intindex=attrIndexMap.get(atrribute);//存放属性的各个值和分类值LinkedHashSetvalues=newLinkedHashSet();LinkedHashSetclasses=newLinkedHashSet();for(inti=0;i

7、s.add(deData[i][class_index]);}}values_count=values.size();class_count=classes.size();intvalues_vector[]=newint[values_count*class_count];intclass_vector[]=newint[class_count];for(inti=0;i

8、)){break;}else{j++;}}intk=