python实现K近邻算法(KNN)

k近邻算法

k近邻法是一种基本分类与回归方法

k近邻法输入为实例的特征向量,对应于特征空间的点;输出为实例的类别,可以取多类(与感知机不同的点之一)。分类时,对新的实例,根据其k个最近邻的训练实例的类别,通过多数表决等方式进行预测。

算法的基本步骤如下:

1)计算已知类别数据集中的点与当前点之间的距离;
2)按照距离递增次序排序;
3)选取与当前点距离最小的k个点;
4)确定前k个点所在类别的出现频率;
5)返回前k个点出现频率最高的类别作为当前点的预测分类。
当然,如果数据集非常庞大的时候,直接计算数据集中的点与当前点之间的距离就会使效率变得非常的低,这里我们通常使用 kd树 来进行优化以便于提高效率。

python实现knn的简单分类

敲代码之前,大家可以简单学一些numpy广播的相关知识numpy广播
import numpy as np


# KNN : k Nearest Neighbors
class k_nn:
    def knnPredict(self, newInput, dataset, labels, k):
        dataset = np.array(dataset)
        # 表示样本数量
        datasize = dataset.shape[0]
        # 求解欧式距离(类似于均方差)(x^2 + y^2)^0.5
        # tile属于numpy模块下边的函数
        # tile(A, reps)返回一个shape=reps的矩阵,矩阵的每个元素是A
        # 比如 A=[0,1,2] 那么,tile(A, 2)= [0, 1, 2, 0, 1, 2]
        # tile(A,(2,2)) = [[0, 1, 2, 0, 1, 2],
        #                  [0, 1, 2, 0, 1, 2]]
        # tile(A,(2,1,2)) = [[[0, 1, 2, 0, 1, 2]],
        #                    [[0, 1, 2, 0, 1, 2]]]
        # 上边那个结果的分开理解就是:
        # 最外层是2个元素,即最外边的[]中包含2个元素,类似于[C,D],而此处的C=D,因为是复制出来的
        # 然后C包含1个元素,即C=[E],同理D=[E]
        # 最后E包含2个元素,即E=[F,G],此处F=G,因为是复制出来的
        # F就是A了,基础元素
        # 综合起来就是(2,1,2)= [C, C] = [[E], [E]] = [[[F, F]], [[F, F]]] = [[[A, A]], [[A, A]]]
        # 这个地方就是为了把输入的测试样本扩展为和dataset的shape一样,然后就可以直接做矩阵减法了。
        # 比如,dataset有4个样本,就是4*2的矩阵,输入测试样本肯定是一个了,就是1*2,为了计算输入样本与训练样本的距离
        # 那么,需要对这个数据进行作差。这是一次比较,因为训练样本有n个,那么就要进行n次比较;
        # 为了方便计算,把输入样本复制n次,然后直接与训练样本作矩阵差运算,就可以一次性比较了n个样本。
        # 比如inX = [0,1],dataset就用函数返回的结果,那么
        # tile(inX, (4,1))= [[ 0.0, 1.0],
        #                    [ 0.0, 1.0],
        #                    [ 0.0, 1.0],
        #                    [ 0.0, 1.0]]
        # 作差之后
        # diffMat = [[-1.0,-0.1],
        #            [-1.0, 0.0],
        #            [ 0.0, 1.0],
        #            [ 0.0, 0.9]]
        diffMat = np.tile(newInput, (datasize, 1)) - dataset
        sqDisMat = diffMat ** 2
        # 对横轴进行求和
        sqDistance = np.sum(sqDisMat, axis=1)
        # 开根号
        distance = sqDistance ** 0.5
        # 对距离进行升序排序,返回的是数组下标
        sortDistance = distance.argsort()

        # 字典结构,存放对应的投票数
        classcount = {}

        # 投票过程
        for i in range(k):
            voteLabel = labels[sortDistance[i]]
            # 如果不存在这个类,则默认为0,每次都加1
            classcount[voteLabel] = classcount.get(voteLabel, 0) + 1

        # 返回一个降序的列表
        Realresult = sorted(classcount.items(), key=lambda item: item[1], reverse=True)
        return Realresult[0][0]


if __name__ == "__main__":
    model = k_nn()
    group = np.array([
        [1.0, 0.9],
        [1.0, 1.0],
        [0.1, 0.2],
        [0.0, 0.1]
    ])
    labels = ['A', 'A', 'B', 'B']
    newInput = [1.2, 1.0]
    print('the class of test sample is ' + model.knnPredict(newInput, group, labels, 3))

python实现knn的简单分类进阶版

import numpy as np


# KNN : k Nearest Neighbors
class k_nn_s:
    def knnPredict(self, newInput, dataset, labels, k):
        dataset = np.array(dataset)
        # 表示样本数量
        datasize = dataset.shape[0]
        # 求解欧式距离(类似于均方差)(x^2 + y^2)^0.5
        # tile属于numpy模块下边的函数
        # tile(A, reps)返回一个shape=reps的矩阵,矩阵的每个元素是A
        # 比如 A=[0,1,2] 那么,tile(A, 2)= [0, 1, 2, 0, 1, 2]
        # tile(A,(2,2)) = [[0, 1, 2, 0, 1, 2],
        #                  [0, 1, 2, 0, 1, 2]]
        # tile(A,(2,1,2)) = [[[0, 1, 2, 0, 1, 2]],
        #                    [[0, 1, 2, 0, 1, 2]]]
        # 上边那个结果的分开理解就是:
        # 最外层是2个元素,即最外边的[]中包含2个元素,类似于[C,D],而此处的C=D,因为是复制出来的
        # 然后C包含1个元素,即C=[E],同理D=[E]
        # 最后E包含2个元素,即E=[F,G],此处F=G,因为是复制出来的
        # F就是A了,基础元素
        # 综合起来就是(2,1,2)= [C, C] = [[E], [E]] = [[[F, F]], [[F, F]]] = [[[A, A]], [[A, A]]]
        # 这个地方就是为了把输入的测试样本扩展为和dataset的shape一样,然后就可以直接做矩阵减法了。
        # 比如,dataset有4个样本,就是4*2的矩阵,输入测试样本肯定是一个了,就是1*2,为了计算输入样本与训练样本的距离
        # 那么,需要对这个数据进行作差。这是一次比较,因为训练样本有n个,那么就要进行n次比较;
        # 为了方便计算,把输入样本复制n次,然后直接与训练样本作矩阵差运算,就可以一次性比较了n个样本。
        # 比如inX = [0,1],dataset就用函数返回的结果,那么
        # tile(inX, (4,1))= [[ 0.0, 1.0],
        #                    [ 0.0, 1.0],
        #                    [ 0.0, 1.0],
        #                    [ 0.0, 1.0]]
        # 作差之后
        # diffMat = [[-1.0,-0.1],
        #            [-1.0, 0.0],
        #            [ 0.0, 1.0],
        #            [ 0.0, 0.9]]
        diffMat = np.tile(newInput, (datasize, 1)) - dataset
        sqDisMat = diffMat ** 2
        # 对横轴进行求和
        sqDistance = np.sum(sqDisMat, axis=1)
        # 开根号
        distance = sqDistance ** 0.5
        # 对距离进行升序排序,返回的是数组下标
        sortDistance = distance.argsort()

        # 字典结构,存放对应的投票数
        classcount = {}

        # 投票过程
        for i in range(k):
            voteLabel = labels[sortDistance[i]]
            # 如果不存在这个类,则默认为0,每次都加1
            classcount[voteLabel] = classcount.get(voteLabel, 0) + 1

        # 返回一个降序的列表
        Realresult = sorted(classcount.items(), key=lambda item: item[1], reverse=True)
        return Realresult[0][0]


# 读取文件数据
def filedata(filename):
    fr = open("./Data/knn_data.txt")
    num = len(fr.readlines())
    # 创建一个num行2列的矩阵
    returnMat = np.zeros((num, 2), dtype=float)

    fr = open("./Data/knn_data.txt")
    count = 0
    # 储存数据类型
    dataLabels = []
    for line in fr.readlines():
        # 去掉换行符
        line = line.strip()
        # 默认逗号隔开
        line = line.split()
        returnMat[count, :] = line[0:2]
        dataLabels.append(line[2])
        count += 1
    return returnMat, dataLabels


# 数据归一化
def dataNormal(dataset):
    # 横轴方向最小的两个数据,
    # 例如:
    # group = np.array([
    #     [1.0, 0.9],
    #     [1.0, 1.0],
    #     [0.1, 1.2],
    #     [0.0, 100]
    # ])
    # minVals = group.min(0)
    # <<<[0.  0.9]
    mindata = dataset.min(0)
    maxdata = dataset.max(0)
    # 两数相减(对应元素相减)
    sub = maxdata - mindata
    m = dataset.shape[0]
    # 让训练数据对应位置减去最小数据
    resultData = dataset - np.tile(mindata, (m, 1))
    # 作商
    resultData = resultData / np.tile(sub, (m, 1))
    return resultData


# 数据测试(简单交叉验证)
def dataTest():
    model = k_nn_s()
    # 90%数据用作训练,10%数据用作测试
    hoRatio = 0.10
    returnMat, dataLabels = filedata('./Data/knn_data.txt')
    # 将文件中的数据广播化
    returnMat = np.array(returnMat)
    dataLabels = np.array(dataLabels)
    resultData = dataNormal(returnMat)
    m = resultData.shape[0]
    numTestval = int(m * hoRatio)
    errorcount = 0.0
    for i in range(numTestval):
        classifierResult = model.knnPredict(resultData[i, :],
                                            resultData[numTestval:m, :], dataLabels[numTestval:m], 3)
        print("这个样例返回的是: " + classifierResult + ", 真正的分类是: " + dataLabels[i])
        if classifierResult != dataLabels[i]:
            errorcount += 1.0
    print("测试总数为 : " + str(numTestval) + ", 分类错误的数量为 : " + str(errorcount))
    print("分类错误的概率为 : " + str(errorcount / float(numTestval)))
    # 结果为:
    # 这个样例返回的是: A, 真正的分类是: A
    # 测试总数为 : 1, 分类错误的数量为: 0.0
    # 分类错误的概率为 : 0.0


if __name__ == "__main__":
    dataTest()

进阶版主要添加了文件读取,数据归一化和测试(基本交叉检验)的操作
文件中的内容为

在这里插入图片描述

未完待续!!!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值