BaggingClassifier意外关键字参数'max_depth'

Question

我在scikit-learn 0.22.2.post1中遇到BaggingClassifier错误。我正在使用python 3.8.2。

from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import BaggingClassifier
bag_clf = BaggingClassifier(
                            DecisionTreeClassifier(
                                                    random_state=0,
                                                    criterion='entropy'
                                                    ),
                              n_estimators=100,
                              max_samples=100,
                              max_depth=2,
                              bootstrap=True,
                              random_state=0
                              )

TypeError：init（）获得了意外的关键字参数'max_depth'

如果删除max_depth=2，则可以从代码中创建对象。 max_depth=2是我得到错误的唯一参数。

任何人都知道这是怎么回事吗？

Answer 1

max_depth是DecisionTreeClassifier（docs）的自变量，而不是BaggingClassifier（docs）的自变量；您应该将定义更改为

bag_clf = BaggingClassifier(
                            DecisionTreeClassifier( max_depth=2,
                                                    random_state=0,
                                                    criterion='entropy'
                                                    ),
                              n_estimators=100,
                              max_samples=100,
                              bootstrap=True,
                              random_state=0
                              )