当前位置：网站首页>基于sklearn的决策树应用实战

基于sklearn的决策树应用实战

2022-08-10 07:59:00 【白话机器学习】

在sklearn中，我们主要使用两个决策树方法，均封装在tree模块下。

分别是分类树tree.DecisionTreeClassifier、回归树tree.DecisionTreeRegressor，对应分别基于决策树解决分类问题和回归问题。

本次主要以分类树的使用过程为例进行介绍。在sklearn中分类决策树的函数方法如下：

DecisionTreeClassifier(class_weight=None, criterion='gini', max_depth=None,
                       max_features=None, max_leaf_nodes=None,
                       min_impurity_decrease=0.0, min_impurity_split=None,
                       min_samples_leaf=1, min_samples_split=2,
                       min_weight_fraction_leaf=0.0, presort=False,
                       random_state=None, splitter='best')

主要参数介绍：
1、criterion
这个参数是用来指定构造树时，树的“不纯度”的计算方法，主要有两种：
（1）entropy，使用信息熵来构造
（2）gini，使用基尼系数来构造（默认采用此模式）

原网站

版权声明
本文为[白话机器学习]所创，转载请带上原文链接，感谢
https://blog.csdn.net/zy_dreamer/article/details/126230369

边栏推荐

猜你喜欢

随机推荐