python 西瓜书决策树实现(可以直接运行)
在西瓜数据集2.0上基于信息增益准则生成决策树。用python 写一个递归函数就可以实现了。 西瓜数据集2.0 # -*- coding: utf-8 -*- ###以西瓜书76为例决策树 import pandas as pd import numpy as np import math import copy #计算信息熵 def Information_Entropy(plist): ent = 0 for p in plist: ent += p*math.log(p,2) ent = -ent return ent def get_p(
下载地址
用户评论