fit（）得到意外的关键字参数“criterian”

Question

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

我做了以下事情：

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

但它给出了一个错误：

^{pr2}$

它在默认参数'gini'下工作正常，但不适用于此参数。在

Answer 1

你可能想

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