python机器学习——KNN(K近邻算法)

背景与原理:

KNN算法其实是逻辑最简单的分类算法——我们认为一个数据的类型是由与其最接近的数据决定的,而“接近”实际上就是我们度量两个数据点之间的距离,如果我们把一组数据看做一个向量$(x_{1},...,x_{n},y)$,其中$y$代表这个数据的类别,那么两组数据$X_{i},X_{j}$间的距离如果使用欧式距离表示为$L_{ij}=\sqrt{\sum_{k=1}^{n}(x_{ik}-x_{jk})^{2}}$

那么对于空间中已经分好类的若干数据,如果我们想预测输入的某组数据属于什么类,那么我们去计算空间中分好类的数据点离它的距离,然后选择距离最近的认为和想预测的数据是最接近的,直接选择距离最近的数据点的类别即可。

当然,只选择最近的是由很大问题的,所以K近邻算法(K-Nearest Neighbours)就是将上述“最近”改成“K个最近的”,然后找出这K个数据点中概率最大的类别作为我们预测的类别。

代码实现:

import numpy as np
import math
from scipy import stats
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

def dis(alpha,beta):
    s=0
    for i in range(0,len(alpha)-1):
        s+=(alpha[i]-beta[i])**2

    return math.sqrt(s)

def KNN_Test(input,dataset,siz,K):
    q=[]
    p=[]
    for i in range(0,siz):
        d=dis(input,dataset[i])
        maxp=0
        for j in range(0,len(q)):
            if q[j]>q[0]:
                maxp=j

        if len(q)<K:
            q.append(d)
            p.append(i)
        elif q[maxp]>d:
            q[maxp]=d
            p[maxp]=i
    cnt=0
    for i in range(0,K):
        if dataset[p[i]][2]==1:
            cnt+=1
    if cnt<K/2:
        return 0
    else:
        return 1


x=np.arange(0.,10.,0.02)
y=5-2*x/3+np.random.randn(500)
now=0
dataset=[]
for i in range(0,500):
    typ = 0
    if 2*x[i]+3*y[i] <= 15:
        if abs(np.random.randn(1)[0])<2:
            typ = 1
        else:
            typ = 0
    else:
        if abs(np.random.randn(1)[0]) < 2:
            typ = 0
        else:
            typ = 1

    dataset.append([x[i],y[i],typ])

c_cnt=0
for i in range(0,500):
    predict_typ=KNN_Test(dataset[i],dataset,500,10)
    if predict_typ==dataset[i][2]:
        c_cnt+=1

print(c_cnt/500)

这段代码生成的分类数据和逻辑回归的相同,我们这次使用KNN算法进行分类,准确率相对可观

小结与改进:

KNN算法存在相当的不足之处,在改进中我们可以选择加入权值——虽然我们考虑最近的K个“邻居”,但是我们认为一组数据肯定与离它最近的邻居更接近,因此在计算概率时我们对于距离较近的数据点给出一个较高的权值,距离较远的邻居给出一个较低的权值,这样能更好地体现“近邻”这个观点。

同样,究竟选择几个“邻居”也是值得考虑的,这一点可以通过权衡样本数量来进行调整。

KNN算法的另一个缺点在于训练样本数量很大、维度很多的时候计算开销很大,因此我们可以在预处理阶段使用降维算法对维度进行简化以提高效率

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章