ValueError: multi_class must be in (‘ovo’, ‘ovr’)

2024年6月8日2024年6月17日

roc_auc_score関数を使用してROC AUCスコアを計算しようとすると、multi_classパラメータが必要です。このパラメータは、多クラス分類の場合に使用され、どのように多クラスを処理するかを指定します。

roc_auc_score関数のドキュメントによると、multi_classパラメータは次の値を取ることができます。

• 'ovr’（One-vs-Rest）：このオプションを選択すると、多クラス分類問題が2クラスの問題として処理されます。各クラスについて、そのクラスと他のすべてのクラスを区別する1つの分類器が作成されます。
• 'ovo’（One-vs-One）：このオプションを選択すると、各クラスのすべてのペアの分類器が作成されます。それぞれの分類器は、2つのクラスの間で区別を行います。

どちらのオプションを選択するかは、多クラス分類問題の性質によって異なります。一般的には、クラスの数が多い場合は 'ovo’、クラスの数が少ない場合は 'ovr’ を選択する傾向があります。

したがって、roc_auc_score関数の呼び出し時にmulti_classパラメータを指定することで、エラーを解決することができます。例えば、以下のようになります。

python
roc_auc = roc_auc_score(y_test, y_pred_lgbm, multi_class='ovr')