【问题标题】:fit() got an unexpected keyword argument 'criterion'fit() 得到了一个意外的关键字参数“标准”
【发布时间】: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。

【问题讨论】:

    标签: python machine-learning scikit-learn decision-tree


    【解决方案1】:

    你可能想要

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

    【讨论】:

    • 哦,我太愚蠢了。谢谢你成功了。那么我在声明模型时是否要传递其他参数?
    • 其实你可以查阅文档来检查哪个参数用于哪个函数(或者在输入函数名后直接在Jupyter中使用Shift+Tab)。你在构造函数的signature 中看到的参数应该是在构造函数中传递的。 fit 接受其他参数(sample_weight、check_input、X_idx_sorted)。
    猜你喜欢
    • 1970-01-01
    • 2021-10-23
    • 1970-01-01
    • 2020-06-21
    • 2016-09-17
    • 2015-06-08
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多