Skip to main content
 首页 » 编程设计

python之GridSearchCV,数据框每个部分中每个类的表示

2025年05月04日23bonelee

我必须进行多类分类 (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 时,我们没有得到每个部分的每个类。您是否为每个部分中代表的每个类都提供了解决方案?谢谢。

请您参考如下方法:

不,这是来自您传递给 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

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