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()