我正在使用 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'
可以正常工作,但不能使用这个。
最佳答案
你可能想要
model = DecisionTreeClassifier(criterion="entropy")
model.fit(X_train, y_train)
关于python - fit() 得到了一个意外的关键字参数 'criterion',我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46628837/