From f175eca4d4d9f9fe7b015739ceb33c12e161b3ce Mon Sep 17 00:00:00 2001
From: LufsX <lufsx@outlook.com>
Date: Wed, 13 Nov 2024 17:00:21 +0800
Subject: [PATCH] feat: add test_labels check

---
 public/challenge/svm/main.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/public/challenge/svm/main.py b/public/challenge/svm/main.py
index 2a2e777..1d5a0ec 100644
--- a/public/challenge/svm/main.py
+++ b/public/challenge/svm/main.py
@@ -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()