fix: test when total not match actually total

2024-10-25 23:28:16 +08:00
parent fc59f4f1d8
commit deadf110aa
3 changed files with 3 additions and 3 deletions
@@ -61,7 +61,7 @@ def evaluate_knn(
    temp_log_details = []

    try:
-        for i in range(total):
+        for i in range(len(test_data)):
            # 获取测试数据中的每个样本
            result = user_func(test_data[i], train_data, train_labels)
            if result == test_labels[i]: