用python实现id3_python实现ID3决策树算法

python实现ID3决策树算法

发布时间：2020-08-21 22:02:22

来源：脚本之家

阅读：66

作者：杨柳岸晓风

ID3决策树是以信息增益作为决策标准的一种贪心决策树算法

# -*- coding: utf-8 -*-

from numpy import *

import math

import copy

import cPickle as pickle

class ID3DTree(object):

def __init__(self): # 构造方法

self.tree = {} # 生成树

self.dataSet = [] # 数据集

self.labels = [] # 标签集

# 数据导入函数

def loadDataSet(self, path, labels):

recordList = []

fp = open(path, "rb") # 读取文件内容

content = fp.read()

fp.close()

rowList = content.splitlines() # 按行转换为一维表

recordList = [row.split("\t") for row in rowList if row.strip()] # strip()函数删除空格、Tab等

self.dataSet = recordList

self.labels = labels

# 执行决策树函数

def train(self):