• 第三节--k近邻算法


    第三节–k近邻算法

    k近邻法(k-nearest neighbor,KNN)是一种基本分类与回归方法.k近邻法的输入为实例的特征变量,对应于特征空间的点,输出为实例的类别,可以取多类.k近邻法假设给定一个训练数据集,其中的实例类别已定,分类时,对新的实例,根据其k个最近邻的训练实例的类别,通过多数表决等方式进行预测.因此,k近邻法不具有显式的学习过程,k近邻法实际上利用训练数据集对特征向量空间进行划分,并作为其分类的"模型".k值的选择,距离度量分类决策规则是k近邻法的三个基本要素

    首先叙述k近邻算法,然后讨论k近邻法的模型及三个基本要素,最后讲述k近邻法的一个实现方法–kd树

    一.k近邻算法

    k近邻算法简单,直观:给定一个训练数据集,对新的输入实例,在训练数据集中找到与该实例最邻近的k个实例,这k个实例的多树属于某个类,就把该输入实例分为这个类

    输入:训练数据集
    T={(x1,y1),(x2,y2), ,(xN,yN)} T=left{left(x_{1}, y_{1} ight),left(x_{2}, y_{2} ight), cdots,left(x_{N}, y_{N} ight) ight}

    其中,xiXRnx_{i} in mathcal{X} subseteq mathbf{R}^{n}为实例的特征向量,yiY={c1,c2, ,cK}y_{i} in mathcal{Y}=left{c_{1}, c_{2}, cdots, c_{K} ight}为实例的类别,i=1,2, ,Ni=1,2, cdots, N;实例特征向量x

    输出:实例x所属的类y

    1. 根据给定的距离度量,在训练集T中找出与x最近邻的k个点,涵盖这k个点的x的邻域记作为Nk(x)N_{k}(x)
    2. Nk(x)N_{k}(x)中根据分类决策规则(如多数表决)决定x的类别y:
      y=argmaxcjxiNk(x)I(yi=cj),i=1,2, ,N,j=1,2, ,K y=arg max _{c_{j}} sum_{x_{i} in N_{k}(x)} Ileft(y_{i}=c_{j} ight),i=1,2, cdots, N, quad j=1,2, cdots, K
      其中I为指示函数,即当yi=cjy_{i}=c_{j}时I为1,否则I为0

    k近邻法的特殊情况是k=1的情形,称为最近邻算法,对于输入的实例点(特征向量)x,最近邻法将训练数据集中与x最近邻点的类作为x的类

    k近邻法没有显式的学习过程

    二.k近邻模型

    k近邻法使用的模型实际上对应于对特征空间的划分,模型由三个基本要素—距离度量,k值的选择分类决策规则决定

    1.模型

    k近邻法中,当训练集,距离度量(如欧式距离),k值及分类决策规则(如多数表决)确定后.对于任何一个新的输入实例,它所属的类唯一地确定.这相当于根据上述要素将特征空间划分为一些子空间,确定子空间里的每个点所属的类,这一事实从最近邻算法中可以看得很清楚

    特征空间中,对每个训练实例点xix_{i},距离该点比其他点更近的所有点组成一个区域,叫作单元(cell).每个训练实例点拥有一个单元,所有训练实例点的单元构成对特征空间的一个划分,最近邻法将实例xix_{i}的类yiy_{i}作为其单元中所有点的类标记(class label).这样,每个单元的实例点的类别是确定的

    from IPython.display import Image
    Image(filename="./data/3_1.png",width=500)
    

    在这里插入图片描述

    2.距离度量

    特征空间中两个实例点的距离是两个实例点相似程度的反映.k近邻模型的特征空间一般是n维实数向量空间Rnmathbf{R}^{n}.使用的距离是欧式距离,但也可以是其他距离.如更一般的LpL_{p}距离(LpL_{p} distance)或Minkowski距离(Minkowski distance)

    设特征空间Xmathcal{X}是n维实数向量空间Rnmathbf{R}^{n},xi,xjX,xi=(xi(1),xi(2), ,xi(n)),xj=(xj(1),xj(2), ,xj(n))Tx_{i}, x_{j} in mathcal{X}, quad x_{i}=left(x_{i}^{(1)}, x_{i}^{(2)}, cdots, x_{i}^{(n)} ight)^{ op},x_{j}=left(x_{j}^{(1)}, x_{j}^{(2)}, cdots, x_{j}^{(n)} ight)^{mathrm{T}},xi,xjx_{i}, x_{j}LpL_{p}距离定义为:
    Lp(xi,xj)=(i=1nxi(l)xj(l)p)1p L_{p}left(x_{i}, x_{j} ight)=left(sum_{i=1}^{n}left|x_{i}^{(l)}-x_{j}^{(l)} ight|^{p} ight)^{frac{1}{p}}

    这里p≥1.当p=2时,称为欧式距离(Euclidean distance),即:
    L2(xi,xj)=(i=1nxi(l)xj(l)2)12 L_{2}left(x_{i}, x_{j} ight)=left(sum_{i=1}^{n}left|x_{i}^{(l)}-x_{j}^{(l)} ight|^{2} ight)^{frac{1}{2}}

    当p=1时,称为曼哈顿距离(Manhattan distance),即:
    L1(xi,xj)=l=1nxi(l)xj(l) L_{1}left(x_{i}, x_{j} ight)=sum_{l=1}^{n}left|x_{i}^{(l)}-x_{j}^{(l)} ight|

    p=p=infty时,称为闵式距离(Minkowski distance),它是各个坐标距离的最大值,即:
    L(xi,xj)=maxlxi(l)xj(l) L_{infty}left(x_{i}, x_{j} ight)=max _{l}left|x_{i}^{(l)}-x_{j}^{(l)} ight|

    下图给出了二维空间中p取不同值时,与原点的LpL_{p}距离为1(LpL_{p}=1)的点的图片

    Image(filename="./data/3_2.png",width=500)
    

    在这里插入图片描述

    下面的例子说明,由不同的距离度量所确定的最近邻点是不同的

    实例1:已知二维空间的3个点x1=(1,1)T,x2=(5,1)T,x3=(4,4)Tx_{1}=(1,1)^{mathrm{T}}, x_{2}=(5,1)^{mathrm{T}}, x_{3}=(4,4)^{mathrm{T}},试求在p取不同值时,LpL_{p}距离下x1x_{1}的最近邻点

    Image(filename="./data/3_3.png",width=500)
    

    在这里插入图片描述

    于是得到:p等于1或2时,x2x_{2}x1x_{1}的最近邻点;p大于等于3时,x3x_{3}x1x_{1}的最近邻点

    3.k值的选择

    k值的选择会对k近邻法的结果产生重大影响

    如果选择较小的k值,就相当于用较小的邻域中的训练实例进行预测,"学习"的近似误差(approximation error)会减小,只有与输入实例较近的训练实例才会对预测结果其作用,但缺点是"学习"的估计误差(estimation error)会增大,预测结果会近邻的实例点非常敏感.如果邻近的实例点恰巧是噪声.预测就会出错.换句话说,k值的减小就意味着整体模型变得复杂,容易发生过拟合

    如果选择较大的k值,就相当于用较大邻域中的训练实例进行预测,其优点是可以减少学习的估计误差,但缺点是学习的近似误差会增大.这时与输入实例较远训练实例也会对预测其作用,使预测发生错误,k值的增大就意味着整体的模型变得简单

    4.分类决策规则

    k近邻法中的分类决策规则往往是多数表决,即由输入实例的k个邻近的训练实例中的多数类决定输入实例的类

    多数表决规则(majority voting rule)有如下解释:如果分类的损失函数为0-1损失函数,分类函数为:
    f:Rn{c1,c2, ,cK} f : mathbf{R}^{n} ightarrowleft{c_{1}, c_{2}, cdots, c_{K} ight}

    那么误分类的概率是:
    P(Yf(X))=1P(Y=f(X)) P(Y eq f(X))=1-P(Y=f(X))

    对给定的实例xXx in mathcal{X},其最近邻的k个训练实例点构成集合Nk(x)N_{k}(x).如果涵盖Nk(x)N_{k}(x)的区域的类别是cjc_{j},那么误分类率是:
    1kxiNk(x)I(yicj)=11kxiNk(x)I(yi=cj) frac{1}{k} sum_{x_{i} in N_{k}(x)} Ileft(y_{i} eq c_{j} ight)=1-frac{1}{k} sum_{x_{i} in N_{k}(x)} Ileft(y_{i}=c_{j} ight)

    要使误分类率最小即检验风险最小,就要使xiNk(x)I(yi=cj)sum_{x_{i} in N_{k}(x)} Ileft(y_{i}=c_{j} ight)最大,所以多数表决规则等价于经验风险最小化

    三.k近邻法的实现:kd树

    实现k近邻法时,主要考虑的问题是如何对训练数据进行快速k近邻搜索.这点在特征空间的维数大及训练数据容量大时尤其必要

    k近邻法最简单的实现方法是线性扫描(linear scan).这时要计算输入实例与每一个训练实例的距离.当训练集很大时,计算非常耗时,这种方法是不可行的

    为了提高k近邻搜索的效率,可以考虑使用特殊的结构存储训练数据,以减少计算距离的次数.kd树(kd tree)方法就是一种

    1.构造kd树

    kd树是一种对k维空间中的实例点进行存储以便对其进行快速检索的树形数据结构.kd树是二叉树,表示对k维空间的一个划分(partition).构造kd树相当于不断地用垂直于坐标轴的超平面将k维空间划分,构成一系列的k维超矩形区域.kd树的每个结点对应于一个k维矩形区域

    构造kd树的方法如下:构造根结点,使根结点对应于k维空间中包含所有实例点的超矩形区域;通过下面的递归方法,不断地对k维空间进行切分,生成子结点,在超矩形区域(结点)上选择一个坐标轴和在此坐标轴上的一个切分点,确定一个超平面,这个超平面通过选定的切分点并垂直于选定的坐标轴.将当前超矩形区域切分为左右两个子区域(子结点);这时实例被分到两个子区域,这个过程直到子区域内没有实例时停止(终止时的结点为叶结点).在此过程中,将实例保存在相应的结点上

    通常,依次选择坐标轴对空间切分.选择训练实例点在选定坐标轴上的中位数(median)为切分点,这样得到的kd树是平衡的.注意,平衡的kd树搜索时的效率未必是最优的

    构造平衡kd树

    输入:k维空间数据集T={x1,x2, ,xN}T=left{x_{1}, x_{2}, cdots, x_{N} ight}
    其中xi=(xi(0),xi(2), ,xi(k))T,i=1,2, ,Nx_{i}=left(x_{i}^{(0)}, x_{i}^{(2)}, cdots, x_{i}^{(k)} ight)^{mathrm{T}}, quad i=1,2, cdots, N

    输出:kd树

    1. 开始:构造根结点,根结点对应于包含TT的k维空间的超矩形区域
      选择x(1)x^{(1)}为坐标轴,以TT中所有实例的x(1)x^{(1)}的中位数为切分点,将根结点对应的超矩形区域切分为两个子区域.切分由通过切分点并与坐标轴x(1)x^{(1)}垂直的超平面

    由根结点生成深度为1的左右子结点;左子结点对应坐标x(1)x^{(1)}小于切分点的子区域,右子结点对应于坐标x(1)x^{(1)}大于切分点的子区域

    将落在切分超平面上的实例点保存在根结点

    1. 重复:对深度为j的结点,选择x(l)x^{(l)}为切分的坐标轴,l=j( mod k)+1l=j(mod k)+1以该结点的区域中所有实例的x(l)x^{(l)}坐标的中位数为切分点,将该结点对应的超矩形区域切分为两个子区域,切分由通过切分点并与坐标轴x(l)x^{(l)}垂直的超平面实现

    由该结点生成深度为j+1的左右子结点:左子结点对应坐标x(l)x^{(l)}小于切分点的子区域,右子结点对应坐标x(l)x^{(l)}大于切分点的子区域

    将落在切分超平面上的实例点保存在该结点

    1. 直到两个子区域没有实例存在时停止,从而形成kd树的区域划分

    实例2:给定一个二维空间的数据集
    T={(2,3)T,(5,4)T,(9,6)T,(4,7)T,(8,1)T,(7,2)T} T=left{(2,3)^{mathrm{T}},(5,4)^{mathrm{T}},(9,6)^{mathrm{T}},(4,7)^{mathrm{T}},(8,1)^{mathrm{T}},(7,2)^{mathrm{T}} ight}
    构造一个平衡kd树

    解:根结点对应包含数据集TT的矩形,选择x(1)x^{(1)}轴,6个数据点的x(1)x^{(1)}坐标的中位数是7(注意:2,4,5,7,8,9在数学中的中位数为6,但因该算法的中值需在点集合之内,所以中值计算用的是len(points)/2=3,points[3]=(7,2)),以平面x(1)=7x^{(1)}=7将空间分为左右两个子矩形(子结点);接着,左矩形以x(2)=4x^{(2)}=4分为两个子矩形,右矩形以x(2)=6x^{(2)}=6分为两个子矩形,如此递归,最后得到如图所示的特征空间划分和kd树

    Image(filename="./data/3_5.png",width=500)
    

    在这里插入图片描述

    Image(filename="./data/3_4.png",width=500)
    

    在这里插入图片描述

    2.搜索kd树

    利用kd树可以省去对大部分数据点的搜索,从而减少搜索的计算量,这里以最近邻为例加以叙述,同样的方法可以应用到k近邻

    给定一个目标点,搜索其最近邻.首先找到包含目标点的叶结点;然后从该叶结点出发,依次回退到父结点;不断查找与目标点最近邻的结点,当确定不可能存在更近的结点时终止,这样搜索就被限制在空间的局部区域上,效率大为提高

    用kd树的最近邻搜索

    输入:已构造的kd树,目标点x
    输出:x的最近邻

    1. 在kd树中找出包含目标点x的叶结点;从根结点出发,递归地向下访问kd树,若目标点x当前维的坐标小于切分点的坐标,则移动左子结点,否则移动到右子结点,直到子结点为叶节点为止

    2. 以此叶节点为"当前最近点"

    3. 递归地向上回退,在每个节点进行一下操作
      a.如果该结点保存的实例点比当前最近点距离目标点更近,则以该实例点为"当前最近点"
      b.当前最近点一定存在于该结点一个子结点对应的区域.检查该子结点的父结点的另一个结点对应的区域是否有更近的点.具体地,检查另一子结点对应的区域是否与以目标点为球心,以目标点与"当前最近点"间的距离为半径的超球体相交
      如果相交,可能在另一个子结点对应的区域内存在距目标点更近的点,移动到另一个子结点.接着递归地进行最近邻搜索
      如果不想交,向上回退

    4. 当回退到根结点时,搜索结束,最后的"当前最近点"即为x的最近邻点
      如果实例点是随机分布的,kd树搜索的平均计算复杂度是O(logN)O(log N)

    实例:给定一个如下图所示的kd树,根节点为A,其子结点为B,C等,树上共存储7个实例点;另一个输入目标实例点S,求S的最近邻

    Image(filename="./data/3_8.png",width=500)
    

    在这里插入图片描述

    首先在kd树中找到包含点S的叶结点D(图中的右下区域),以点D作为近似最近邻.真正最近邻一定在以点S为中心通过点D的圆的内部.然后返回结点D的父结点B,在结点B的另一子结点F的区域内搜索最近邻,结点F的区域与圆不想交,不可能有最近邻点,继续返回上一级父结点A,在结点A的另一子结点C的区域内搜索最近邻,结点C的区域与圆相交;该区域在园内的实例点有点E,点E比点D更近,成为新的最近邻近似,最后得到点E是点S的最近邻

    四.代码实现

    1.度量距离

    import math
    from itertools import combinations
    
    # p=1 Manhattan distance
    # p=2 Euclidean distance
    # p=3 Minkowski distance
    def L(x,y,p=2):
        # x1=[1,1] x2=[5,1]
        if len(x)==len(y) and len(x)>1:
            sum=0
            for i in range(len(x)):
                sum+=math.pow(abs(x[i]-y[i]),p)
            return math.pow(sum,1/p)
        else:
            return 0
    
    # 实例1
    x1=[1,1]
    x2=[5,1]
    x3=[4,4]
    
    # x1,x3
    for i in range(1,5):
        r={"1-{}".format(c):L(x1,c,p=i) for c in [x2,x3]}
        print(min(zip(r.values(),r.keys())))
    
    (4.0, '1-[5, 1]')
    (4.0, '1-[5, 1]')
    (3.7797631496846193, '1-[4, 4]')
    (3.5676213450081633, '1-[4, 4]')
    

    2.自定义KNN分析iris

    遍历所有数据点,找出n个距离最近的点的分类情况,少数服从多数

    %matplotlib inline
    import numpy as np
    import pandas as pd
    import matplotlib.pyplot as plt
    
    from sklearn.datasets import load_iris
    from sklearn.model_selection import train_test_split
    
    from collections import Counter
    
    iris=load_iris()
    df=pd.DataFrame(iris.data,columns=iris.feature_names)
    df["label"]=iris.target
    df.columns=["sepal length","sepal width","petal length","petal width","label"]
    # data=np.array(df.iloc[:100,[0,1,-1]])
    
    df
    
    sepal length sepal width petal length petal width label
    0 5.1 3.5 1.4 0.2 0
    1 4.9 3.0 1.4 0.2 0
    2 4.7 3.2 1.3 0.2 0
    3 4.6 3.1 1.5 0.2 0
    4 5.0 3.6 1.4 0.2 0
    5 5.4 3.9 1.7 0.4 0
    6 4.6 3.4 1.4 0.3 0
    7 5.0 3.4 1.5 0.2 0
    8 4.4 2.9 1.4 0.2 0
    9 4.9 3.1 1.5 0.1 0
    10 5.4 3.7 1.5 0.2 0
    11 4.8 3.4 1.6 0.2 0
    12 4.8 3.0 1.4 0.1 0
    13 4.3 3.0 1.1 0.1 0
    14 5.8 4.0 1.2 0.2 0
    15 5.7 4.4 1.5 0.4 0
    16 5.4 3.9 1.3 0.4 0
    17 5.1 3.5 1.4 0.3 0
    18 5.7 3.8 1.7 0.3 0
    19 5.1 3.8 1.5 0.3 0
    20 5.4 3.4 1.7 0.2 0
    21 5.1 3.7 1.5 0.4 0
    22 4.6 3.6 1.0 0.2 0
    23 5.1 3.3 1.7 0.5 0
    24 4.8 3.4 1.9 0.2 0
    25 5.0 3.0 1.6 0.2 0
    26 5.0 3.4 1.6 0.4 0
    27 5.2 3.5 1.5 0.2 0
    28 5.2 3.4 1.4 0.2 0
    29 4.7 3.2 1.6 0.2 0
    ... ... ... ... ... ...
    120 6.9 3.2 5.7 2.3 2
    121 5.6 2.8 4.9 2.0 2
    122 7.7 2.8 6.7 2.0 2
    123 6.3 2.7 4.9 1.8 2
    124 6.7 3.3 5.7 2.1 2
    125 7.2 3.2 6.0 1.8 2
    126 6.2 2.8 4.8 1.8 2
    127 6.1 3.0 4.9 1.8 2
    128 6.4 2.8 5.6 2.1 2
    129 7.2 3.0 5.8 1.6 2
    130 7.4 2.8 6.1 1.9 2
    131 7.9 3.8 6.4 2.0 2
    132 6.4 2.8 5.6 2.2 2
    133 6.3 2.8 5.1 1.5 2
    134 6.1 2.6 5.6 1.4 2
    135 7.7 3.0 6.1 2.3 2
    136 6.3 3.4 5.6 2.4 2
    137 6.4 3.1 5.5 1.8 2
    138 6.0 3.0 4.8 1.8 2
    139 6.9 3.1 5.4 2.1 2
    140 6.7 3.1 5.6 2.4 2
    141 6.9 3.1 5.1 2.3 2
    142 5.8 2.7 5.1 1.9 2
    143 6.8 3.2 5.9 2.3 2
    144 6.7 3.3 5.7 2.5 2
    145 6.7 3.0 5.2 2.3 2
    146 6.3 2.5 5.0 1.9 2
    147 6.5 3.0 5.2 2.0 2
    148 6.2 3.4 5.4 2.3 2
    149 5.9 3.0 5.1 1.8 2

    150 rows × 5 columns

    plt.scatter(df[:50]["sepal length"],df[:50]["sepal width"],label="0")
    plt.scatter(df[50:100]["sepal length"],df[50:100]["sepal width"],label="1")
    plt.xlabel("sepal length")
    plt.ylabel("sepal width")
    plt.legend()
    
    <matplotlib.legend.Legend at 0x18193faffd0>
    

    在这里插入图片描述

    data=np.array(df.iloc[:100,[0,1,-1]])
    X,y=data[:,:-1],data[:,-1]
    X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.2)
    
    class KNN(object):
        def __init__(self,X_train,y_train,n_neighbors=3,p=2):
            """
            parameter:n_neighbors 临近点个数
            parameter:p 距离度量
            """
            self.n=n_neighbors
            self.p=p
            self.X_train=X_train
            self.y_train=y_train
            
        def predict(self,X):
            # 取出n个点
            knn_list=[]
            for i in range(self.n):
                dist=np.linalg.norm(X-self.X_train[i],ord=self.p)
                knn_list.append((dist,self.y_train[i]))
                
            for i in range(self.n,len(self.X_train)):
                max_index=knn_list.index(max(knn_list,key=lambda x:x[0]))
                dist=np.linalg.norm(X-self.X_train[i],ord=self.p)
                if knn_list[max_index][0]>dist:
                    knn_list[max_index]=(dist,self.y_train[i])
                    
            # 统计
            knn=[k[-1] for k in knn_list]
            count_pairs=Counter(knn)
            max_count=sorted(count_pairs,key=lambda x:x)[-1]
            return max_count
        
        def score(self,X_test,y_test):
            right_count=0
            n=10
            for X,y in zip(X_test,y_test):
                label=self.predict(X)
                if label==y:
                    right_count+=1
            return right_count/len(X_test)
    
    clf=KNN(X_train,y_train)
    
    clf.score(X_test,y_test)
    
    1.0
    
    test_point=[6.0,3.0]
    print("Test Point:{}".format(clf.predict(test_point)))
    
    Test Point:1.0
    
    plt.scatter(df[:50]["sepal length"],df[:50]["sepal width"],label="0")
    plt.scatter(df[50:100]["sepal length"],df[50:100]["sepal width"],label="1")
    plt.plot(test_point[0],test_point[1],"bo",label="test_point")
    plt.xlabel("sepal length")
    plt.ylabel("sepal width")
    plt.legend()
    
    <matplotlib.legend.Legend at 0x181944c5588>
    

    在这里插入图片描述

    3.sklearn实现KNN

    from sklearn.neighbors import KNeighborsClassifier
    
    clf_sk=KNeighborsClassifier()
    clf_sk.fit(X_train,y_train)
    
    KNeighborsClassifier(algorithm='auto', leaf_size=30, metric='minkowski',
               metric_params=None, n_jobs=None, n_neighbors=5, p=2,
               weights='uniform')
    
    clf_sk.score(X_test,y_test)
    
    1.0
    

    sklearn.neighbors.KNeighborsClassifier主要参数说明:

    • n_neighbors:临近点个数
    • p:度量距离
    • algorithm:近邻算法,可选{“auto”,“ball_tree”,“kd_tree”,“brute”}
    • weights:确定近邻的权重

    4.kd树

    # kd-tree每个结点中主要包含的数据结构如下 
    class KdNode(object):
        def __init__(self, dom_elt, split, left, right):
            self.dom_elt = dom_elt  # k维向量节点(k维空间中的一个样本点)
            self.split = split      # 整数(进行分割维度的序号)
            self.left = left        # 该结点分割超平面左子空间构成的kd-tree
            self.right = right      # 该结点分割超平面右子空间构成的kd-tree
     
     
    class KdTree(object):
        def __init__(self, data):
            k = len(data[0])  # 数据维度
            
            def CreateNode(split, data_set): # 按第split维划分数据集exset创建KdNode
                if not data_set:    # 数据集为空
                    return None
                # key参数的值为一个函数,此函数只有一个参数且返回一个值用来进行比较
                # operator模块提供的itemgetter函数用于获取对象的哪些维的数据,参数为需要获取的数据在对象中的序号
                #data_set.sort(key=itemgetter(split)) # 按要进行分割的那一维数据排序
                data_set.sort(key=lambda x: x[split])
                split_pos = len(data_set) // 2      # //为Python中的整数除法
                median = data_set[split_pos]        # 中位数分割点             
                split_next = (split + 1) % k        # cycle coordinates
                
                # 递归的创建kd树
                return KdNode(median, split, 
                              CreateNode(split_next, data_set[:split_pos]),     # 创建左子树
                              CreateNode(split_next, data_set[split_pos + 1:])) # 创建右子树
                                    
            self.root = CreateNode(0, data)         # 从第0维分量开始构建kd树,返回根节点
    
    
    # KDTree的前序遍历
    def preorder(root):  
        print (root.dom_elt)  
        if root.left:      # 节点不为空
            preorder(root.left)  
        if root.right:  
            preorder(root.right)
    
    # 对构建好的kd树进行搜索,寻找与目标点最近的样本点:
    from math import sqrt
    from collections import namedtuple
    
    # 定义一个namedtuple,分别存放最近坐标点、最近距离和访问过的节点数
    result = namedtuple("Result_tuple", "nearest_point  nearest_dist  nodes_visited")
      
    def find_nearest(tree, point):
        k = len(point) # 数据维度
        def travel(kd_node, target, max_dist):
            if kd_node is None:     
                return result([0] * k, float("inf"), 0) # python中用float("inf")和float("-inf")表示正负无穷
     
            nodes_visited = 1
            
            s = kd_node.split        # 进行分割的维度
            pivot = kd_node.dom_elt  # 进行分割的“轴”
            
            if target[s] <= pivot[s]:           # 如果目标点第s维小于分割轴的对应值(目标离左子树更近)
                nearer_node  = kd_node.left     # 下一个访问节点为左子树根节点
                further_node = kd_node.right    # 同时记录下右子树
            else:                               # 目标离右子树更近
                nearer_node  = kd_node.right    # 下一个访问节点为右子树根节点
                further_node = kd_node.left
     
            temp1 = travel(nearer_node, target, max_dist)  # 进行遍历找到包含目标点的区域
            
            nearest = temp1.nearest_point       # 以此叶结点作为“当前最近点”
            dist = temp1.nearest_dist           # 更新最近距离
            
            nodes_visited += temp1.nodes_visited  
     
            if dist < max_dist:     
                max_dist = dist    # 最近点将在以目标点为球心,max_dist为半径的超球体内
                
            temp_dist = abs(pivot[s] - target[s])    # 第s维上目标点与分割超平面的距离
            if  max_dist < temp_dist:                # 判断超球体是否与超平面相交
                return result(nearest, dist, nodes_visited) # 不相交则可以直接返回,不用继续判断
                
            #----------------------------------------------------------------------  
            # 计算目标点与分割点的欧氏距离  
            temp_dist = sqrt(sum((p1 - p2) ** 2 for p1, p2 in zip(pivot, target)))     
            
            if temp_dist < dist:         # 如果“更近”
                nearest = pivot          # 更新最近点
                dist = temp_dist         # 更新最近距离
                max_dist = dist          # 更新超球体半径
            
            # 检查另一个子结点对应的区域是否有更近的点
            temp2 = travel(further_node, target, max_dist) 
            
            nodes_visited += temp2.nodes_visited
            if temp2.nearest_dist < dist:        # 如果另一个子结点内存在更近距离
                nearest = temp2.nearest_point    # 更新最近点
                dist = temp2.nearest_dist        # 更新最近距离
     
            return result(nearest, dist, nodes_visited)
     
        return travel(tree.root, point, float("inf"))  # 从根节点开始递归
    
    # 实例2
    data = [[2,3],[5,4],[9,6],[4,7],[8,1],[7,2]]
    kd = KdTree(data)
    preorder(kd.root)
    
    [7, 2]
    [5, 4]
    [2, 3]
    [4, 7]
    [9, 6]
    [8, 1]
    
    from time import clock
    from random import random
    
    # 产生一个k维随机向量,每维分量值在0~1之间
    def random_point(k):
        return [random() for _ in range(k)]
     
    # 产生n个k维随机向量 
    def random_points(k, n):
        return [random_point(k) for _ in range(n)]
    
    ret = find_nearest(kd, [3,4.5])
    print (ret)
    
    Result_tuple(nearest_point=[2, 3], nearest_dist=1.8027756377319946, nodes_visited=4)
    
    N = 400000
    t0 = clock()
    kd2 = KdTree(random_points(3, N))            # 构建包含四十万个3维空间样本点的kd树
    ret2 = find_nearest(kd2, [0.1,0.5,0.8])      # 四十万个样本点中寻找离目标最近的点
    t1 = clock()
    print ("time: ",t1-t0, "s")
    print (ret2)
    
    E:Anacondaenvsmytensorflowlibsite-packagesipykernel_launcher.py:2: DeprecationWarning: time.clock has been deprecated in Python 3.3 and will be removed from Python 3.8: use time.perf_counter or time.process_time instead
      
    
    
    time:  6.159827752999263 s
    Result_tuple(nearest_point=[0.09732020950704356, 0.49930092577904095, 0.8029864162744909], nearest_dist=0.004072918366121865, nodes_visited=42)
    
    
    E:Anacondaenvsmytensorflowlibsite-packagesipykernel_launcher.py:5: DeprecationWarning: time.clock has been deprecated in Python 3.3 and will be removed from Python 3.8: use time.perf_counter or time.process_time instead
      """
    
  • 相关阅读:
    解决 Windows 资源管理器 CPU 占用率很高的问题
    彻底卸载Cygwin
    Cygwin 快速镜像站点
    解决 cygwin 在 win10 下中文显示乱码的问题
    酷派大神 F1 连接调试,日志获取的方法
    陌路
    游戏开发中自定义脚本与配置
    在 cygwin 的环境下调用 libgdx gdx-tools
    在 cygwin 的环境下使用 git 的时候中文编码的问题
    对 android apk 进行重新签名操作
  • 原文地址:https://www.cnblogs.com/LQ6H/p/10744783.html
Copyright © 2020-2023  润新知