机器学习实战-sorted()函数

最新推荐文章于 2024-01-24 20:07:53 发布

原创最新推荐文章于 2024-01-24 20:07:53 发布 · 475 阅读

0 ·

CC 4.0 BY-SA版权

Py机器学习实战专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个简单的多数类别计数算法实现，该算法通过遍历输入的类别列表并使用字典来统计每个类别的出现次数，最后返回出现次数最多的类别。此算法常用于决策树、K近邻等机器学习算法中作为分类决定的依据。

def majorityCnt(classList):
    classCount={}
    for vote in classList:
        if vote not in classCount.keys(): classCount[vote] = 0
        classCount[vote] += 1
    sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True)
    #第一个参数是classCount的迭代器对象（类似列表），第二个参数是按照第1个域值排序，第三个参数是按降序排序
    return sortedClassCount[0][0]