fit() 得到了一个意外的关键字参数“标准”

Posted 2023-03-12

【中文标题】fit() 得到了一个意外的关键字参数“标准”

【英文标题】：fit() got an unexpected keyword argument 'criterion'

【发布时间】：2018-03-19 14:57:57

【问题描述】：

我正在使用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' 可以正常工作，但不能使用 this。

【参考方案1】：

你可能想要

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

【讨论】：

哦，我太愚蠢了。谢谢你成功了。那么我在声明模型时是否要传递其他参数？

其实你可以查阅文档来检查哪个参数用于哪个函数（或者在输入函数名后直接在Jupyter中使用Shift+Tab）。你在构造函数的signature 中看到的参数应该是在构造函数中传递的。 fit 接受其他参数（sample_weight、check_input、X_idx_sorted）。