机器学习实战-决策树

决策树的第一个算法为,计算给定数据集的香农熵 香农熵的计算公式都给出来了,下面贴一下代码 还是比较好懂的,就是先创建一个字典,在字典里面对每个类别新建一个键值对,值为每个类别的样本个数 之后分别计算概率啊什么的 输出如下 跟书上的答案是一样的 我个人有一点点的疑问是对字典类型的自建函数 下面看看python文档 dict.keys()              返回一个包含字典中键的列表 >>>
相关文章
相关标签/搜索