diff --git a/public/experiment/logreg/main.py b/public/experiment/logreg/main.py index 67eeffa..f4011ed 100644 --- a/public/experiment/logreg/main.py +++ b/public/experiment/logreg/main.py @@ -66,7 +66,7 @@ def evaluate_logreg( total = 68 # LogReg 测试样本数 temp_log_details = [] - if len(test_data) > total: + if len(test_data) != total: log_details.append("测试集样本数不匹配,请检查 load_data 函数!") print("测试集样本数不匹配,请检查 load_data 函数!") log_details.append("成绩为零!")