更新 function.py

function.py

@@ -5,7 +5,7 @@ import numpy as np
 import os
 from collections import Counter
 from sklearn.preprocessing import MinMaxScaler
-from sklearn.neighbors import KDTree
+from sklearn.neighbors import KDTree  # 正确导入 KDTree
 
 # 加载数据集
 def load_dataset(directory):
@@ -59,7 +59,7 @@ def main():
     test_data = normalize_data(test_data)
 
     # 使用 KDTree 构建训练集的树结构
-    tree = KDTree(training_data)
+    tree = KDTree(training_data)  # 构建 KDTree
 
     # 测试 kNN 算法
     total_tests = len(test_data)