网站首页 文章专栏 使用sklearn计算向量距离 & 最近邻 & smote
使用sklearn计算向量距离 & 最近邻 & smote
创建于:2019-05-01 07:26:53 更新于:2021-10-16 03:35:54 羽瀚尘 429
python python,机器学习



计算最近邻

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).

Reference


[给出了包含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/)