main.asv

来自「KNN最近邻分类器和预测器」· ASV 代码 · 共 27 行

K=9;
[train_set,test_set]=extract(train);
train_size=size(train_set,1)
test_size=size(test_set,1)
column=size(train_set,2)
correct_predict_num=0;
for i=1:test_size;
    for j=1:train_set  
        dis(i)=distance(test_set(i,:),train_set(j,:));
    end
    [sorted_dis,index]=sort(dis);
    k=0;
    for m=1:9
        if train_set(m,column)==1
            k++;
    end
    if k>4
        class=1;
    else
        class=0;
    end
    if class==test_set(i,column)
        correct_predict_num=correct_predict_num+1;
end
ration