【发布时间】: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