完整報錯為:ValueError: Target is multiclass but average='binary'. Please choose another average setting, one of [None, 'micro', 'macro', 'weighted'].
解決方法
對于多分類任務,將
from sklearn.metrics import f1_score f1_score(y_test, y_pred)
改為:f1_score(y_test, y_pred,average='macro') # average按任務需要計算
?