决策树⑤——Python代码实现决策树

最新推荐文章于 2024-03-13 16:41:30 发布

VIP文章数据小斑马

最新推荐文章于 2024-03-13 16:41:30 发布

阅读量3.6k

点赞数 5

分类专栏：决策树文章标签：决策树python实现 python代码生成决策树决策树生成范例决策树例子决策树分类树

本文链接：https://blog.csdn.net/cindy407/article/details/93324285

版权

决策树系列目录（文末有大礼相送）：
决策树①——信息熵&信息增益&基尼系数
 决策树②——决策树算法原理（ID3，C4.5，CART）
决策树③——决策树参数介绍（分类和回归）
决策树④——决策树Sklearn调参（GridSearchCV）
决策树应用实例①——泰坦尼克号分类
 决策树应用实例②——用户流失预测模型
 决策树应用实例③——银行借贷模型
 决策树应用实例④——淘宝&京东白条(回归&均方差&随机森林）

上一篇总结了Sklearn库用GridSearchCV调参的过程，今天用python依据现代女生的审美创建一颗简单的决策树，看下是否能代表大众的心声~~

from math import log
import operator


def calcShannonEnt(dataset):
    numEntries = len(dataset)  # 输入样本个数
    labelCounts = {}  # 存类别及相应的数量
    for featVec in dataset:  # 一行即是一个特征向量
        currentLable = featVec[-1] # 取每个样本的类别
        if currentLable not in labelCounts.keys():#