上传者: 38716519
|
上传时间: 2022-11-04 16:03:21
|
文件大小: 94KB
|
文件类型: PDF
上一篇博客主要介绍了决策树的原理,这篇主要介绍他的实现,代码环境python 3.4,实现的是ID3算法,首先为了后面matplotlib的绘图方便,我把原来的中文数据集变成了英文。
原始数据集:
变化后的数据集在程序代码中体现,这就不截图了
构建决策树的代码如下:
#coding :utf-8
'''
2017.6.25 author :Erin
function: decesion tree ID3
'''
import numpy as np
import pandas as pd
from math import log
import operator
def load_