fitness.m

来自「朴素贝叶斯分类器」· M 代码 · 共 14 行

function [y z]=fitness(chrom,traindata1,testdata1)
[m n]=size(chrom);%m是一个染色体的长度，n是种群的大小，即有几个染色体
y=[];z=[];
chrom1=zhuanhua(chrom);
for i=1:n
    h=find(chrom1(:,i)==1);
    traindata=traindata1(h,:);
    [Y C]=fenleiqi(traindata,testdata1);%Y表示经过算法测试样本获得的类的最大概率和；C是分类正确的个数
    y=[y;Y];    %上面的traindata是traindata1的子集
    z=[z;C];
end