AUCを計算

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

パイプラインで定義されたランダムフォレスト分類器を用いて、グリッドサーチを行い、最適なハイパーパラメータを見つけ、テストデータでの予測と評価を行いましたね。では、AUCを計算するためのコードを追加しましょう。

python
# AUCの計算
y_pred_proba_test = best_model.predict_proba(X_test)
fpr, tpr, thresholds = roc_curve(y_test, y_pred_proba_test[:, 1])
auc_score = auc(fpr, tpr)
print("AUC Score:", auc_score)

このコードを元のコードに追加することで、テストデータでのAUCスコアを計算し、出力することができます。