python - GridSearchCV，数据框每个部分中每个类的表示

Question

我必须进行多类分类 (3)。我使用 GridSearchCV 为我的分类器搜索最佳参数。

但我有一个不平衡的 x_train(和 x_test):0 有 3079 个实例，1 有 12 个实例，3 有 121 个实例。

我有这个错误:

Target is multiclass but average='binary'. Please choose another average setting.

我认为这个错误来自拆分，当我们拆分 x_train 时，我们没有得到每个部分的每个类。您是否为每个部分中代表的每个类都提供了解决方案？谢谢。

Answer 1

不，这是来自您传递给 GridSearchCV(....., scoring=SOME_METRIC... ) 的评分函数。请显示该部分。

如果它是其中之一，并且分类任务是多类的，那么它们就无法工作:

• 'f1'
• '记忆'
• '精度'

然后您需要提供 average 参数。

因此，如果您当前使用以下评分:

GridSearchCV(....., scoring='f1'... )

然后您需要像这样附加“微观”、“宏观”或“加权”之一:

GridSearchCV(....., scoring='f1_micro'... )

precision 或 recall 相同

有关详细信息，请参阅此页面:http://scikit-learn.org/stable/modules/model_evaluation.html#the-scoring-parameter-defining-model-evaluation-rules

如果您使用其他东西，请显示代码。