feat: check test item number

2024-11-01 20:09:25 +08:00
parent 41086d29b7
commit e9a8807a6b
3 changed files with 23 additions and 2 deletions
@@ -66,6 +66,13 @@ def evaluate_logreg(
    total = 68  # LogReg 测试样本数
    temp_log_details = []

+    if len(test_data) > total:
+        log_details.append("测试集样本数不匹配，请检查 load_data 函数！")
+        print("测试集样本数不匹配，请检查 load_data 函数！")
+        log_details.append("成绩为零！")
+        print("成绩为零！")
+        return 0.0
+
    try:
        # 训练模型
        time_start = time.time()