TypeError: f1_score() 至少需要 2 个参数（1 个给定）

Question

我正在尝试运行 SGDClassifier，但出现此错误：

TypeError: f1_score() takes at least 2 arguments (1 given)

这是我的代码：

pipeline = Pipeline([
 ('clf', SGDClassifier())
])

parameters = {
    'clf__loss': ('log', 'hinge'),
    'clf__penalty': ['l1', 'l2', 'elasticnet'],
    'clf__alpha': [0.001, 0.0001, 0.00001, 0.000001]
}

score_func = make_scorer(metrics.f1_score(average='weighted'))

grid_search = GridSearchCV(pipeline, parameters, n_jobs=3,
verbose=1, scoring=score_func)



grid_search.fit(X, Y)

我该如何解决？

Answer 1

你需要定义评分函数如下：

from sklearn.metrics import f1_score


f1_scorer = make_scorer(f1_score,average='weighted')

grid_search =GridSearchCV(pipeline, parameters, n_jobs=3,verbose=1, scoring=f1_scorer)

这很好用，可以做你想做的事。

希望这会有所帮助。

Answer 2

我相信你需要像这样通过它：

make_scorer(metrics.f1_score, average='weighted')

您应该单独传递函数名称和任何关联的关键字参数。

Documentation参考。