更新 function.py

2024-10-17 16:14:03 +08:00
parent 2dfff87b47
commit cb8607b935
1 changed files with 1 additions and 1 deletions
@@ -52,7 +52,7 @@ def classify0(inX, dataSet, labels, k=3):
    
    # 计算马氏距离
    diff = normalized_dataSet - normalized_inX
-    distances = np.sqrt(np.sum(diff @ inv_cov_matrix * diff, axis=1))
+    distances = np.einsum('ij,jk->i', diff, inv_cov_matrix @ diff.T)  # 使用 einsum 简化计算
    
    # 获取距离最小的 k 个样本的索引
    sorted_indices = np.argsort(distances)[:k]