信息增益生成决策树_用信息增益法画出决策树

作者：我家小花儿 | 2024-03-22 05:06:37

踩

用信息增益法画出决策树

首先查看数据集

在这里插入图片描述
我们用这个数据集来构造决策树，判断一个新的西瓜是否为好瓜。

决策树的构造

计算数据集的信息熵

首先观察数据集 $D$ ，发现数据集 $D$ 中有好瓜和坏瓜两个类别，其中好瓜占比 $p_1=\frac {8}{17}$ ，坏瓜占比 $p_2=\frac {9}{17}$ ,计算出数据集 $D$ 的信息熵为
$Ent(D)=-\sum_{k=1}^{2}p_klog_2p_k=-(\frac{8}{17}log_2\frac{8}{17}+\frac{9}{17}log_2\frac{9}{17})=0.998$

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/285494