Scikit-Learn 中的决策树
在sklearn
中,决策树在以下位置:
sklearn.tree
主要包含这么几种树:
包 | 作用 |
---|---|
tree.DecisionTreeClassifier |
分类树 |
tree.DecisionTreeRegressor |
回归树 |
tree.export_graphviz |
将生成的决策树导出为 DOT 格式,画图专用 |
tree.ExtraTreeClassifier |
高随机版本的分类树 |
tree.ExtraTreeRegressor |
高随机版本的回归树 |