sklearnのGridSearchCVに指定可能な評価指標

この記事について

• 単なるメモです。
• GridSearchCVのscoringオプションに指定可能な評価指標を確認する方法です。

grid = GridSearchCV(
    model,
    param_grid,
    cv=5,
    scoring="neg_log_loss", #←　★これ★
    verbose=3,
    n_jobs=4
)

利用可能な評価指標一覧の表示

• これで表示できます。

import sklearn
from sklearn import *
from pprint import pprint
pprint(sorted(sklearn.metrics.SCORERS.keys()))

手元の環境だと

['accuracy',
 'adjusted_mutual_info_score',
 'adjusted_rand_score',
 'average_precision',
 'balanced_accuracy',
 'brier_score_loss',
 'completeness_score',
 'explained_variance',
 'f1',
 'f1_macro',
 'f1_micro',
 'f1_samples',
 'f1_weighted',
 'fowlkes_mallows_score',
 'homogeneity_score',
 'jaccard',
 'jaccard_macro',
 'jaccard_micro',
 'jaccard_samples',
 'jaccard_weighted',
 'max_error',
 'mutual_info_score',
 'neg_log_loss',
 'neg_mean_absolute_error',
 'neg_mean_squared_error',
 'neg_mean_squared_log_error',
 'neg_median_absolute_error',
 'normalized_mutual_info_score',
 'precision',
 'precision_macro',
 'precision_micro',
 'precision_samples',
 'precision_weighted',
 'r2',
 'recall',
 'recall_macro',
 'recall_micro',
 'recall_samples',
 'recall_weighted',
 'roc_auc',
 'v_measure_score']