split.m

来自「决策树」· M 代码 · 共 17 行

function class_split_a=split(class,n)
%class_train=zeros(n,1);
num=0;
for i=1:n
    if class(i)==1;
        num=num+1;
    end
end
class_split_a_1=num/n;
class_split_a_2=1-class_split_a_1;
if class_split_a_1>=class_split_a_2
    class_split_a=class_split_a_1;
else
    class_split_a=class_split_a_2;
end