roc_auc_scoreを使用してAUCを計算

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

roc_auc_scoreを使用してAUCを計算しようとしていますね。ただし、この関数は2クラス分類の場合に使用されます。多クラス分類の場合、roc_auc_scoreはオプションとしてmulti_classを設定する必要があります。

ここでは、One-vs-Rest (OvR) スキームを使用して多クラス分類のAUCを計算します。修正されたコードは次のようになります：

python
from sklearn.metrics import roc_auc_score

# AUCの計算
auc_lr = roc_auc_score(y_test, y_pred_lr, multi_class='ovr')  # 多クラスの場合は'multi_class'を指定
print("Logistic Regression AUC:", auc_lr)

これで、ロジスティック回帰モデルのテストデータでのAUCが計算され、出力されるはずです。