python - fit() 得到了一个意外的关键字参数 'criterion'

Question

我正在使用 sklearn.tree.DecisionTreeClassifier here是它的链接。我想使用关键字 criterion 并将其设置为 "entropy"

我做了以下事情:

model = DecisionTreeClassifier()
model.fit(X_train, y_train, criterion = "entropy")

但它给出了这个错误:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-103-509174b2caad> in <module>()
      1 model = DecisionTreeClassifier()
----> 2 model.fit(X_train, y_train, criterion = "entropy")

TypeError: fit() got an unexpected keyword argument 'criterion'

使用默认参数 'gini' 可以正常工作，但不能使用这个。

Answer 1

你可能想要

model = DecisionTreeClassifier(criterion="entropy")
model.fit(X_train, y_train)