matlab K最近邻

Question

我是 matlab 的超级新手。我想实现 KNN 算法。我试图阅读 fitcknn 分类器，但我无法理解。 我有矩阵 x，它有 4 个输入向量（每个向量有 3 个特征）

     1     2     3
     5    19    20
     1     2     4
     8    19    21

我想得到一个输出矩阵 Y，它为输入矩阵的每个向量提供最近的邻居（按顺序）。 例如：y 在这种情况下将是

      3     2     4
      4     3     1
      1     2     4
      2     3     1

说明：矩阵Y的第一行表示离向量1最近的向量是：向量3然后向量2然后向量4。

是否有库可以进行这种分类（使用余弦距离作为相似度函数）？ 谢谢。

Answer 1

n = size(x,1);
dist = squareform(pdist(x,'cosine')); %// distance matrix
dist(1:n+1:end) = inf; %// self-distance doesn't count
[~, y] = sort(dist,2);
y = y(:,1:n-1);

---

为了节省内存，您可以使用pdist2 而不是pdist 分块工作：

n = size(x,1);
m = 100; %// chunk size. As large as memory allows. Divisor of n
y = NaN(n,n-1); %// pre-allocate results
for ii = 0:m:size(x,1)-1
    ind = ii+(1:m); %// current chunk: these rows
    dist_chunk = pdist2(x(ind,:),x,'cosine'); %// results for this chunk
    [~, y_chunk] = sort(dist_chunk,2);
    y(ind,:) = y_chunk(:,2:end); %// fill results, except self-distance
end