更新 function.py

function.py

@@ -58,11 +58,17 @@ def classify(input_vector, dataSet, labels, k=3):
     """
     使用欧式距离的 kNN 分类算法。
     """
-     # 计算欧式距离
-    distances = np.linalg.norm(dataSet - input_vector, axis=1)
+     # 计算协方差矩阵及其逆矩阵
+    cov_matrix = np.cov(dataSet, rowvar=False)
+    inv_cov_matrix = np.linalg.inv(cov_matrix)
     
+    # 计算马氏距离
+    diffMat = dataSet - input_vector
+    left_term = np.dot(diffMat, inv_cov_matrix)
+    mahalanobis_distances = np.sqrt(np.sum(left_term * diffMat, axis=1))
+
     # 获取距离最近的 k 个样本的索引
-    k_indices = np.argsort(distances)[:k]
+    k_indices = np.argsort(mahalanobis_distances)[:k]
     
     # 获取最近邻样本的标签
     k_nearest_labels = [labels[i] for i in k_indices]