Update app.py

app.py

@@ -92,12 +92,17 @@ if uploaded_file is not None:
 
     # ROC 曲線
     # 假設 2 是正標籤
-    fpr_stack, tpr_stack, _ = roc_curve(y_test, y_pred_stack_proba, pos_label=2)
+   # 將 y_test 標籤轉換為 0 和 1
+    y_test_binary = (y_test == 2).astype(int)  # 假設 2 是正標籤
+
+    # 計算 ROC 曲線
+    fpr_stack, tpr_stack, _ = roc_curve(y_test_binary, y_pred_stack_proba)
     roc_auc_stack = auc(fpr_stack, tpr_stack)
 
-    fpr_vote, tpr_vote, _ = roc_curve(y_test, y_pred_vote_proba, pos_label=2)
+    fpr_vote, tpr_vote, _ = roc_curve(y_test_binary, y_pred_vote_proba)
     roc_auc_vote = auc(fpr_vote, tpr_vote)
 
+
     fig, ax = plt.subplots()
     ax.plot(fpr_stack, tpr_stack, color='blue', lw=2, label='堆疊分類器 (AUC = %0.2f)' % roc_auc_stack)
     ax.plot(fpr_vote, tpr_vote, color='red', lw=2, label='投票分類器 (AUC = %0.2f)' % roc_auc_vote)