开发者:上海品职教育科技有限公司 隐私政策详情

应用版本:4.2.11(IOS)|3.2.5(安卓)APP下载

Martinmm0305 · 2024年01月19日

College-degree 的info gain跟答案算出的不一样。

 27:51 (1.3X) 

我算出的college degree 的weighted avg Gini 是0.7509, 则根据公式得出的info gain 是0.971 - 0.7509 = 0.22,

我是按照car owner的例子算的,是我哪里算错了吗?



1 个答案
已采纳答案

李坏_品职助教 · 2024年01月19日

嗨,努力学习的PZer你好:


info gain的意思是entropy下降的程度,也就是从base entropy到weighted entropy下降了多少。


根据前面Car owner的计算过程,我来写一下College degree的计算过程:

  1. 在持有大学学历的4个样本中,0个人made claim。所以持有大学学历的entropy = 0.
  2. 在没有大学学历的6个样本中,4个人made claim,所以没有大学学历的entropy = -[4/6*log2(4/6) + 2/6*log2(2/6)] = 0.918.
  3. weighted entropy = 4/10 * 0 + 6/10*0.918 = 0.551.
  4. info gain = base entropy - weighted entropy = 0.971-0.551 = 0.42

----------------------------------------------
虽然现在很辛苦,但努力过的感觉真的很好,加油!

Martinmm0305 · 2024年01月20日

多谢!

  • 1

    回答
  • 0

    关注
  • 133

    浏览
相关问题