您的位置:首页 > 其它

kdtree 计算数据与样本间的相似度

2019-07-11 17:23 78 查看
版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。 本文链接:https://blog.csdn.net/qq_43270687/article/details/95496395
#特征值需要哑编码
import numpy as np
from sklearn.neighbors import KDTree

X = np.array([[0,1,1],
[0,1,0],
[0,0,0],
[1,1,1]])
tree = KDTree(X, leaf_size=2)
dist, ind = tree.query(X[:1], k=4)
print(X)
print(ind)
# ind = indices of 4 closest neighbors
print(dist)
# dist = distances to 4 closest neighbors

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: