在 GridSearchCV 中使用精度作为评分时如何指定正标签

Posted 2023-03-12

【中文标题】在 GridSearchCV 中使用精度作为评分时如何指定正标签

【英文标题】：How to specify positive label when use precision as scoring in GridSearchCV

【发布时间】：2018-11-28 17:19:29

【问题描述】：

model = sklearn.model_selection.GridSearchCV(
        estimator = est, 
        param_grid = param_grid,
        scoring = 'precision',
        verbose = 1,
        n_jobs = 1,
        iid = True,
        cv = 3)

在sklearn.metrics.precision_score(y, y_pred,pos_label=[0]) 中，我可以指定正标签，如何在 GridSearchCV 中也指定？

如果没有办法指定，在使用自定义评分时，我该如何定义？

我试过这个：

custom_score = make_scorer(precision_score(y, y_pred,pos_label=[0]),  
                          greater_is_better=True)  

但我得到了错误：

NameError: name 'y_pred' is not defined

【参考方案1】：

读取docs，您可以将任何kwargs 传递给make_scorer，它们将自动传递给score_func 可调用对象。

from sklearn.metrics import precision_score, make_scorer
custom_scorer = make_scorer(precision_score, greater_is_better=True,  pos_label=0)

然后你将这个custom_scorer 传递给GridSearchCV：

gs = GridSearchCV(est, ..., scoring=custom_scorer)