GridSearchCV调优XGBoost二分类时,精确率/召回率对应哪类?
关于GridSearchCV二分类评分指标对应类别的说明
当你在二分类任务中给GridSearchCV传入'recall'或'precision'作为字符串型评分指标时,默认计算的是**类别1(正类)**的精确率/召回率。
背后的逻辑:
- sklearn中,二分类任务的默认正类(
pos_label)是标签值里数值更大的那个。对于常见的0/1二分类标签,就是类别1;如果是字符串标签(比如['neg', 'pos']),则取字典序靠后的那个。 - 用字符串指标时,GridSearchCV底层调用的是对应metrics函数的默认参数:比如
recall_score(y_true, y_pred, average='binary', pos_label=1),所以最终输出的单一数值就是正类(1)的指标值。
验证方法:
你可以手动调用sklearn.metrics.recall_score或precision_score,使用默认参数计算,结果会和GridSearchCV输出的评分完全一致,以此确认对应类别。
内容的提问来源于stack exchange,提问作者thileepan
相关产品推荐
相关产品推荐

