网站首页 文章专栏 使用sklearn计算向量距离 & 最近邻 & smote
python
kneighbor = 10
from sklearn.neighbors import NearestNeighbors
neighbors = NearestNeighbors(n_neighbors=kneighbor).fit(X)
location_set = neighbors.kneighbors([X[1]], return_distance=False)[0]
可以看到, NearestNeighbors对数据进行计算，并通过neighbors.kneighbors返回某一个向量的最近邻。
neighbors.kneighbors返回的是一个二维数组，这里只计算了X[1]的最近邻，所以函数返回时用[0].
如果希望对所有的数据都计算一个最近邻，那么用neighbors.kneighbors(X, return_distance=False).
[给出了包含smote在内的代码]（http://shataowei.com/2017/12/01/python%E5%BC%80%E5%8F%91%EF%BC%9A%E7%89%B9%E5%BE%81%E5%B7%A5%E7%A8%8B%E4%BB%A3%E7%A0%81%E6%A8%A1%E7%89%88-%E4%B8%80/）