sklearn 决策树(分类树、回归树)的重要参数、属性、方法理解

VariableX

于 2020-07-07 19:43:34 发布

阅读量4.4k

点赞数 11

CC 4.0 BY-SA版权

分类专栏：机器学习基础文章标签：决策树机器学习 python

本文链接：https://blog.youkuaiyun.com/VariableX/article/details/107188730

本文详细介绍了sklearn决策树的重要参数，包括分类树的criterion（推荐设为'gini'）、splitter（推荐设为'best'）、max_depth、min_samples_leaf、min_samples_split、max_features（推荐设为'log2'）和class_weight（推荐设为'balanced'），以及回归树的criterion（默认设为'mse'）。同时讨论了各参数的适用场景和调整建议。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

决策分类树

引入

from sklearn.tree import DecisionTreeClassifier
# 全部参数
DecisionTreeClassifier(criterion='gini',splitter='best',max_depth=None, \
                       min_samples_split=2,min_samples_leaf=1, \
                       min_weight_fraction_leaf=0.0,max_features=None, \
                       random_state=None,max_leaf_nodes=None, \
                       min_impurity_dec