《机器学习实战》ch2源码解析

分类:首先将测试向量inX(1*dims)与训练样本数据dataSet(n*dims)中的每个向量进行余弦距离计算,然后将结果排序,输出topk个最大值;相对这k个lable进行投票累加,累加值最大的就是分类结果;

python对矩阵运算能力比较强大,所以dims维单向量特征的测试向量与训练data的余弦距离是通过矩阵得到的
其中tile(inX, (dataSetSize,1)),将inX转为n*dims矩阵,其中每一列数据就是inX元素….然后一次相减,平方和,开根号得到的sqDistances 就是测试向量与训练data中n个向量的余弦距离
classCount是python中的字典数据类型,(key-value)
sortedClassCount 是通过value值从大到小排序得到的排号序的字典类型

from numpy import *
import operator
from os import listdir
def classify0(inX, dataSet, labels, k):

    dataSetSize = dataSet.shape[0]
    diffMat = tile(inX, (dataSetSize,1)) - dataSet
    sqDiffMat = diffMat**2
    sqDistances = sqDiffMat.sum(axis=1)
    distances = sqDistances**0.5
    sortedDistIndicies = distances.argsort()     
    classCount={}          
    for i in range(k):
        voteIlabel = labels[sortedDistIndicies[i]]
        classCount[voteIlabel] = classCount.get(voteIlabel,0) + 1
    sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True)
    return sortedClassCount[0][0]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值