文章目录
一、sklearn中决策树模块
从sklearn官方文档中决策树官方文档,我们知道所有的Decision Trees算法模块如下:
其具体含义如下所示:
本文主要对决策树模块中的分类树和回归树进行实例讲解。
二、tree.DecisionTreeClassifier分类树
从官方文档我们可以知道tree.DecisionTreeClassifier参数如下所示:
这些参数的含义如下所示:决策树参数、菜菜的sklearn课堂—决策树
其属性和方法如下:
三、tree.DecisionTreeRegressor回归树
从官方文档我们可以知道tree.DecisionTreeRegressor参数如下所示:
对比上面的分类树,我们可以发现回归树几乎所有参数、属性、接口都和分类树一样。需要注意的是在回归树中没有标签分布是否均衡的问题,因此回归树没有class_weight参数,其中criterion取值不一样,具体如下:
四、tree.export_graphviz
从官方文档我们可以知道tree.DecisionTreeRegressor参数如下所示:
参数含义如下:参考链接