auvch / DecisionTreeClassifier Public

Notifications You must be signed in to change notification settings
Fork 0
Star 5

简单的决策树分类器，支持C4.5、CART算法和相应的剪枝算法

5 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
README.md		README.md
main.py		main.py
plotTree.py		plotTree.py
tree.py		tree.py
utils.py		utils.py

Repository files navigation

DecisionTreeClassifier

简单的决策树分类器，支持C4.5、CART算法和相应的剪枝算法

File structure:

utils.py
实现熵、信息增益、信息增益率、基尼系数的的计算
tree.py:
实现决策树类，包括树的生成与剪枝
plotTree.py
实现决策树绘图
main.py
程序入口