数据挖掘 Python实现ID3算法

最新推荐文章于 2024-05-23 09:36:18 发布

Nougats

最新推荐文章于 2024-05-23 09:36:18 发布

阅读量1.3k

点赞数 1

分类专栏： python 文章标签： python 数据挖掘

本文链接：https://blog.csdn.net/Nougats/article/details/72804382

版权

import math

def createDataSet():
    dataSet = [[1, 1, 'yes'], [1, 1, 'yes'], [1, 0, 'no'], [0, 1, 'no'], [0, 1, 'no']]#数据集的最后一列称作value，label称作attr
    labels = ['no surfacing', 'flippers']
    return dataSet, labels
def createTree(dataSet, labels):
    values = [value[-1] for value in dataSet]# 获取当前数据集的value
    if values.count(values[0]) == len(values):#树的叶子节点是同一个类别的数据
        return values[0]
    if len(dataSet[