第1关:什么是决策树
第1题
训练决策树的过程就是构建决策树的过程
ID3算法是根据信息增益来构建决策树
C4.5算法是根据信息增益比来构建决策树
决策树模型的可理解性高
第2题
决策树可以是多叉树。比如买西瓜时有三种情况可供选择:买、不买、待定。此时决策树就不是一棵二叉树。
第2关:信息熵与信息增益
本关任务:掌握什么是信息增益,完成计算信息增益的程序设计。
import numpy as np
def getEnt(array):
Ent = 0
for i in range(2):
Dv = 0