决策树

分类: 文章 • 2024-08-16 18:17:16

from sklearn.tree import DecisionTreeClassifier
tree_clf=DecisionTreeClassifier(max_depth=2)

决策树
调整超参数
max_depth 树的深度
min_samples_split 至少有多少样本数据才进行划分
min_samples_leaf 对于叶子节点来说至少要有几个样本
max_leaf_nodes 最多有多少叶子节点

回归问题

from sklearn.tree import DecisionTreeRegreesor
tree_reg=DecisionTreeRegreesor()

决策树