compare.m

来自「经典教材《机器学习》里的几乎所有算法 作者就是《机器学习》的2大牛人 有的函数添」· M 代码 · 共 24 行

function [rp,rn,pn,pp,GMA,Fp,tp,tn,fp,fn]=compare(test,predict)

tp=0;fn=0;fp=0;tn=0;
for i=1:max(size(test))
    if test(i)==1
        if predict(i)==1
            tp=tp+1;
        else fn=fn+1;
        end
        else 
            if predict(i)==1
                fp=fp+1;
            else
                tn=tn+1;
            end
    end
end
    rn=tn/(tn+fp);
    rp=tp/(tp+fn);
    pn=tn/(tn+fn);
    pp=tp/(tp+fp);
    GMA=sqrt(rn*rp);
    Fp=2*pp*rp/(pp+rp);