From 2f530bd891c72c8cf4da90ff83808cabd9a2d0d2 Mon Sep 17 00:00:00 2001
From: hao <19382120447@163.com>
Date: Tue, 15 Oct 2024 20:01:42 +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 286bb9a..e6d525e 100644
--- a/function.py
+++ b/function.py
@@ -64,7 +64,7 @@ def classify0(inX, dataSet, labels, k=3):
     # 计算待分类样本与训练集中每个样本的马氏距离
     invCovMat = np.linalg.inv(covMat)
     diffVecs = dataSet - meanVec
-    dist = np.sqrt(np.sum(inX - meanVec, axis=1).dot(invCovMat).dot((inX - meanVec).T))
+    dist = np.sqrt(np.sum((inX - meanVec).dot(invCovMat) * (inX - meanVec), axis=1))
     
     # 获取距离最近的 k 个样本的索引
     kNearestIndices = np.argsort(dist)[:k]