sklearn中有关sklearn.tree.DecisionTreeClassifier方法中的参数max_features

来源:5-2 决策树(2)

weixin_慕丝2256177

2022-04-20

老师,sklearn中有关sklearn.tree.DecisionTreeClassifier方法中的参数max_features,是什么意思?如果说我们的数据集中有100个特征,max_features选10,是代表一开始的时候就从100个特征中挑选10个特征来构建一个决策树模型吗?

写回答

1回答

flare_zhao

2022-05-15

max_features限制分枝时考虑的特征个数,超过限制个数的特征都会被舍弃。和max_depth异曲同工, max_features是用来限制高维度数据的过拟合的剪枝参数,但其方法比较暴力,是直接限制可以使用的特征数量而强行使决策树停下的参数,在不知道决策树中的各个特征的重要性的情况下,强行设定这个参数可能会导致模型 学习不足

0
0

Python3入门人工智能 掌握机器学习+深度学习

人工智能基础全方位讲解,构建完整人工智能知识体系,带你入门AI

1946 学习 · 594 问题

查看课程