Update app.py

app.py

@@ -91,10 +91,11 @@ if uploaded_file is not None:
     st.pyplot(fig)
 
     # ROC 曲線
-    fpr_stack, tpr_stack, _ = roc_curve(y_test, y_pred_stack_proba)
+    # 假設 2 是正標籤
+    fpr_stack, tpr_stack, _ = roc_curve(y_test, y_pred_stack_proba, pos_label=2)
     roc_auc_stack = auc(fpr_stack, tpr_stack)
 
-    fpr_vote, tpr_vote, _ = roc_curve(y_test, y_pred_vote_proba)
+    fpr_vote, tpr_vote, _ = roc_curve(y_test, y_pred_vote_proba, pos_label=2)
     roc_auc_vote = auc(fpr_vote, tpr_vote)
 
     fig, ax = plt.subplots()