KNN-KD树

k最近邻(kNN)算法是一种分类与回归的方法,假设我们给定一个searching point,打算在训练数据集中找到一个与该searching point最近的k个数据点,利用多数表决确定这k个点所属的类,将该searching point归入到这个类里面。


kd树是k近邻的一种实现方法:为提高搜索效率

构造kd树

kd树是数据的一种存储结构,是一颗平衡二叉树,同时也是对数据所在空间的划分。就是把整个数据空间划分为几个部分,然后在特定的空间内进行操作。

kd树构造流程

算法直到需要被划分的子树为空是停止。

搜索最近邻:

输入:已经构造的kd树,测试样本x

输出:x的最近邻

a)如果kd树为空,返回null

b)进行二叉查找,在kd树中找到包含x的叶节点,并生成搜索路径:从根节点开始,递归的向下访问kd树。首先将节点压入表示搜索路径的栈中,如果目标点x当前维的坐标小于切分点的坐标,进入左子节点,否则进入右子节点。直到子节点为空。

c)回溯查找:

根据得到的搜索路径栈,栈顶的元素为“当前最近点”,将该元素出栈,并计算该点与x的距离d;

while 栈不空:

          栈顶元素出栈(父节点);

         以x为圆心,d为半径画圆,

         if 这个圆与该元素(父节点)对应的分割超平面相交:

                  计算该元素(父节点)和x的距离,

                           if小于d,则将该元素(父节点)更新为“当前最近点”,d也需要更新;

                  同时对元素(父节点)的另一半子空间的子树进行步骤b),搜索的点加入搜索路径,即压栈操作;

搜索k近邻:

方法:最大堆优先队列寻找k近邻

1)建立一个大小为k的最大堆

2)在寻找最近邻的算法中,每计算一次距离,将对应的样本和距离加入最大堆中,直到最大堆建立完成

3)while 寻找最近邻算法 是否 结束:

             继续寻找最近邻算法,并计算距离

             如果小于堆顶元素:

                          则替换,并维护最大堆

堆中的样本即为x的k个近邻。建立堆的时间是O(k),更新堆的时间为logk,总费时为费时O(k+(n-k)*logk)=O(n*logk)。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 原文章为scikit-learn中"用户指南"-->"监督学习的第六节:Nearest Neighbors"###...
    HabileBadger阅读 11,987评论 0 7
  • B树的定义 一棵m阶的B树满足下列条件: 树中每个结点至多有m个孩子。 除根结点和叶子结点外,其它每个结点至少有m...
    文档随手记阅读 14,565评论 0 25
  • 一些概念 数据结构就是研究数据的逻辑结构和物理结构以及它们之间相互关系,并对这种结构定义相应的运算,而且确保经过这...
    Winterfell_Z阅读 11,304评论 0 13
  • 一.朴素贝叶斯 1.分类理论 朴素贝叶斯是一种基于贝叶斯定理和特征条件独立性假设的多分类的机器学习方法,所...
    wlj1107阅读 8,349评论 0 5
  • 目录 0.树0.1 一般树的定义0.2 二叉树的定义 1.查找树ADT 2.查找树的实现2.1 二叉查找树2.2 ...
    王侦阅读 12,062评论 0 3