JonathonYan1993 / ML_DecisionTree_prepruning_postpruning Public

Notifications You must be signed in to change notification settings
Fork 16
Star 35

本程序实现决策树的建立与可视化，以及决策树的预剪枝与后剪枝，数据集为西瓜书4.2、4.3节中的西瓜数据集

35 stars 16 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
DecisionTree_NoPruning.png		DecisionTree_NoPruning.png
DecisionTree_Postpruning.png		DecisionTree_Postpruning.png
DecisionTree_Prepruning.png		DecisionTree_Prepruning.png
Decision_Tree_ID3.py		Decision_Tree_ID3.py
Decision_Tree_ID3_postpruning.py		Decision_Tree_ID3_postpruning.py
Decision_Tree_ID3_prepruning.py		Decision_Tree_ID3_prepruning.py
Decision_Tree_Visual.py		Decision_Tree_Visual.py
README.md		README.md
watermelon2.csv		watermelon2.csv
watermelon2Training.csv		watermelon2Training.csv
watermelon2Validation.csv		watermelon2Validation.csv
watermelon2_Decision_Tree_ID3.png		watermelon2_Decision_Tree_ID3.png

Repository files navigation

ML_DecisionTree_prepruning_postpruning

本程序基于Numpy实现决策树的建立与可视化，以及决策树的预剪枝与后剪枝，数据集为西瓜书4.2、4.3节中的西瓜数据集

1、repo Structure

Decision_Tree_ID3.py——创建决策树
Decision_Tree_ID3_postpruning.py——决策树后剪枝
Decision_Tree_ID3_prepruning.py——决策树预剪枝
Decision_Tree_Visual.py——决策树可视化

2、数据集

watermelon2.csv
watermelon2Training.csv
watermelon2Validation.csv

3、结果可视化

watermelon2_Decision_Tree_ID3.png等

About

本程序实现决策树的建立与可视化，以及决策树的预剪枝与后剪枝，数据集为西瓜书4.2、4.3节中的西瓜数据集

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%