Skip to content

TheShy321/Decision_Tree

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Decision_tree

本文采用决策树对汽车评价数据集进行测试分析

tree.py文件为建立树的基本模型,里面有三种数据评价的实现方式,里面有详细的注释 分别为信息增益、信息增益率、基尼系数 treePlotter为绘制决策树的函数

main.py为主函数的实现方式,目前已经实现的留出法、 交叉验证的数据集方式等待实现-Come soon

基本逻辑:

留出法 8 - 2 三种决策树算法 ID3 C4.5 Cart

运行方式

点击main.py运行即可。