From cb8607b93563cb512b83e8aed12f72f1ca5c1e35 Mon Sep 17 00:00:00 2001
From: heart20003 <1789588961@qq.com>
Date: Thu, 17 Oct 2024 16:14:03 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/function.py b/function.py
index bb79a49..d0331e4 100644
--- a/function.py
+++ b/function.py
@@ -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]