feat: add test_labels check

2024-11-13 17:00:21 +08:00
parent a3560c974d
commit f175eca4d4
1 changed files with 11 additions and 0 deletions
@@ -73,6 +73,17 @@ def evaluate_svm(
        print("成绩为零！")
        return 0.0

+    if len(set(test_labels)) != 3:
+        log_details.append(
+            "本次 SVM 挑战赛要求不合并死亡与安乐死（即使用 SVM 完成三分类问题），请检查 load_data 函数！"
+        )
+        print(
+            "本次 SVM 挑战赛要求不合并死亡与安乐死（即使用 SVM 完成三分类问题），请检查 load_data 函数！"
+        )
+        log_details.append("成绩为零！")
+        print("成绩为零！")
+        return 0.0
+
    try:
        # 训练模型
        time_start = time.time()