sklearn 中的决策树
一、模块sklearn.tree
sklearn中的决策树都在“tree”这个模块之下，这个模块共包括五个类：
| tree.DecisionTreeClassifier | 分类树 |
| tree.DecisionTreeRegressor | 回归树 |
| tree.export_graphviz | 将生成的决策树导出为DOT格式，画图专用 |
| tree.ExtraTreeClassifier | 高随机版本的分类树 |
| tree.ExtraTreeRegressor | 高随机版本的回归树 |
二、sklearn的基本建模流程
在这个过程中，分类树对应的代码是：
from sklearn import tree # 需要导入的模块
clf = tree.DecisionTreeClassifier() # 实例化
clf = clf.fit(x_train,y_train) # 用训练集数据训练模型
result = clf.score(x_test,y_test) # 导入测试集，从接口中调用需要的信息（根据菜菜的机器学习整理）
4575
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
