代码详解：用Python构建邻域矩阵( 四 )

2019-02-02

寻找邻域

用现代工具寻找邻域是非常简单的。在这里，我们选择使用scipy ，因为稍后将使用这个软件包中的其他工具，但sklearn或其他软件包也可以完成这项工作。使用scipy时，首先使用训练数据集创建一个cKDTree：

from scipy.spatial import cKDTree

tree = cKDTree(XY_train)

可再次查询该tree：

K = 3

result = tree.query(XY_test k=K)

在这里，我们需要测试样本元素中训练样本的三个最近邻域。默认情况下， tree.query返回邻域索引和相关距离。我们将保留两者。

distances indices = result

让我们集中讨论索引数组。

array([[0 1 2

推荐阅读

上一篇：玩转户外直播，除了手机还需要哪些直播设

下一篇：华为仅仅是在手机销量上不如三星？华为与三星的差距大不大？