sklearn笔记：决策树

概述

sklearn中决策树的类都在 tree 这个模块下。这个模块总共包含五个类：

• tree.DecisionTreeClassifier：分类树
• tree.DecisionTreeRegressor：回归树
• tree.export_graphviz：将生成的决策树导出为DOT格式，画图专用
• tree.ExtraTreeClassifier：高随机版本的分类树
• tree.ExtraTreeRegressor：高随机版本的回归树

sklearn基本建模流程：

在这个流程下，分类树对应的代码为：

DecisionTreeClassifier

重要参数

criterion