我正在使用 k-means 做一个项目,我的教授建议使用 kdtree。我在 python 中找到了 kdtree 的实现(我知道 scipy 中也有,但我找不到任何示例实现)。我的问题与标题相同,是使用kdtree来加速k-means,还是我错了?
data = [(2,2),(1,0),(2,3),(10,5),(59,8),(4,2)]
tree = KDTree.construct_from_data(data)
nearest = tree.query(query_point=(5,4), t=3)
print nearest
Run Code Online (Sandbox Code Playgroud)
输出:
[(4, 2), (2, 3), (2, 2)]
Run Code Online (Sandbox Code Playgroud)