Python实现KNN

深藏阁楼爱情的钟 2022-02-28 10:52 187阅读 0赞

# -*- coding: utf-8 -*-
    import numpy as np
    import operator
    
    
    def knnClassify(inX, dataSet, labels, k):
        """ KNN算法 :param inX: 待判断样本的特征 :param dataSet: 数据集各个样本的特征 :param labels: 样本集的标签 :param k: 取前k个结果 """
        dataSetSize = dataSet.shape[0]
        
        # 计算输入的向量inX与所有样本的距离，最小二乘法
        diffMat = np.tile(inX, (dataSetSize, 1)) - dataSet
        sqDiffMat = diffMat ** 2
        sqDistances = sqDiffMat.sum(axis = 1)
        distances = sqDistances ** 0.5
        
        # 对距离大小进行排序
        sortedDistIndices = distances.argsort()
        classCount = { }
        
        # 选择距离最小的 K 个点，以label为键，频数为值
        for i in range(k):
            voteLabel = labels[ sortedDistIndices[i] ]
            classCount[ voteLabel ] = classCount.get(voteLabel, 0) + 1
        
        # 按照类别的数量多少进行排序
        sortedClassCount = sorted(classCount.items(), key=operator.itemgetter(1), reverse=True)  
        print(sortedClassCount[0][0])
        return sortedClassCount[0][0]  # 返回类别数最多的类别名称
        
    
    if __name__ == '__main__':
        inX = np.array([2,2.2,1.9])
        dataSet = np.array([
            [1,1,1],
            [0.9,0.9,0.9],
            [1.2,1,1.1],
            [1.1,0.9,1],
            [2,2.1,2.2],
            [2.2,2.1,2],
            [2.1,2.3,1.9],
            [1.8,2.1,2],
            [3,3.2,3.1],
            [2.9,2.9,3],
            [2.8,3,3.1],
            [3.2,3.1,3]
        ])
        labels = np.array([1,1,1,1,2,2,2,2,3,3,3,3])
        k = 3
        knnClassify(inX, dataSet, labels, k)

发表评论取消回复

表情：

评论列表（有 0 条评论，187人围观）

还没有评论，来说两句吧...

相关阅读

相关 python的knn算法list_Python实现的knn算法示例

本文实例讲述了Python实现的knn算法。分享给大家供大家参考，具体如下：代码参考机器学习实战那本书：机器学习实战 (Peter Harrington著) 中文版

r囧r小猫/ 2022年12月31日 09:23/ 0 赞/ 165 阅读

相关 python实现KNN算法（公式推导+源代码）

今天这篇文章要介绍的是KNN（k近邻算法），这是一种简单的分类算法，它的思想是通过测量不同特征值之间的距离进行分类的。这样说你可能不太懂，那下面我们就通过一个简单的例子来形

ゝ一世哀愁。/ 2022年12月21日 13:54/ 0 赞/ 220 阅读

相关 python实现KNN分类算法

import sklearn import numpy as np from sklearn.neighbors import KNeighborsCl

浅浅的花香味﹌/ 2022年12月20日 11:10/ 0 赞/ 129 阅读

相关 python knn代码详解_具体knn算法概念参考knn代码python实现

具体knn算法概念参考knn代码python实现上面是参考《机器学习实战》的代码，和knn的思想python \ \_\\_ encoding=utf8 \_\\_算法

落日映苍穹つ/ 2022年10月31日 13:46/ 0 赞/ 186 阅读

相关 Knn原理及Python实现、数据展示

KNN原理及优缺点 KNN的python实现 KNN的python代码中的数据格式 KNN原理及优缺点 KNN算法又称为k近邻分类(k-nearest

不念不忘少年蓝@/ 2022年09月25日 05:18/ 0 赞/ 147 阅读

相关 KNN算法的Python实现

最近在看机器学习，第一章就介绍了最简单的KNN算法，一下就是KNN算法的Python实现，要运行这个代码，需要安装Python(x,y)。 from numpy i

叁歲伎倆/ 2022年08月22日 12:26/ 0 赞/ 177 阅读

相关 Python实现KNN算法

Python实现KNN算法 KNN算法的实际用处很多，主要用于分类阶段，是一个基础的分类算法。KNN主要基于距离的计算，一般可以在原始的欧氏空间中计算样本之间的距离。改进版

╰半橙微兮°/ 2022年08月06日 11:10/ 0 赞/ 200 阅读

相关 K近邻算法-KNN-python简单实现

KNN算法是一种简单的机器学习算法，从大致原理上来理解相对简单。主要思路就是计算要分类的测试特征向量与训练数据集的特征向量的距离，计算距离的方法可以有很多种（如欧氏距离，海

女爷i/ 2022年03月17日 14:06/ 0 赞/ 192 阅读

相关 Python实现KNN

-- coding: utf-8 -- import numpy as np import operator def kn

深藏阁楼爱情的钟/ 2022年02月28日 10:52/ 0 赞/ 188 阅读

相关 Python 实现 KNN（K-近邻）算法

Python 实现 KNN（K-近邻）算法一、概述　　KNN（K-最近邻）算法是相对比较简单的机器学习算法之一，它主要用于对事物进行分类。用比较官方的话来说就是：给

曾经终败给现在/ 2022年02月14日 10:05/ 0 赞/ 250 阅读