决策树分类器抛出 KeyError: 'log_loss'

我使用sklearn的决策树，通常有log_loss

classifier = DecisionTreeClassifier(random_state = 42,class_weight ='balanced' ,criterion='log_loss')
classifier.fit(X_train, y_train)

error :

KeyError: 'log_loss'

The log_loss参数的选项criterion仅在最新的 scikit-learn 中添加版本1.1.2:

标准{“基尼”，“熵”，“log_loss”}，默认=“基尼”

前两部都没有这个版本1.0.2 or 版本0.24.2:

标准{“基尼”，“熵”}，默认=“基尼”

该错误表明您使用的是旧版本；你可以检查你的 scikit-learn 版本

import sklearn
print(sklearn.__version__)

因此，您需要将 scikit-learn 升级到 v1.1.2。