from sklearn import datasets
from sklearn.neighbors import KDTree,KNeighborsClassifier,KernelDensity,KNeighborsTransformer
X, y = datasets.load_iris(return_X_y=True)
tree = KDTree(X)
dist_to_knn, idx_of_knn = tree.query(X=X[[3]],k=5)
print(dist_to_knn) # 到k个近邻的距离
print(idx_of_knn) # k个近邻的索引
Sklearn KDTree 的使用
最新推荐文章于 2024-07-21 17:56:10 发布
该博客介绍了如何利用scikit-learn库中的KDTree数据结构来查找数据点的最近邻。通过加载iris数据集并创建KDTree,然后查询第3个样本的最近5个邻居,展示了KDTree在近邻搜索中的应用。输出了距离这些近邻的距离和它们在数据集中的索引。
789

被折叠的 条评论
为什么被折叠?



