knn.m

来自「K近邻法的matlab程序」· M 代码 · 共 12 行

function [label_test] = knn(k, data_train, label_train, data_test)
% knn - k nearest neighbours classifier

error(nargchk(4,4,nargin));

dist = l2_distance(data_train, data_test);
[sorted_dist, nearest] = sort(dist);
nearest = nearest(1:k,:);
label_test = label_train(nearest);

% note this only works for binary labels
label_test = mean(label_test,1) >= 0.5;