更新 function.py

2024-10-15 20:01:42 +08:00
parent 1b93a2e2f0
commit 2f530bd891
1 changed files with 1 additions and 1 deletions
@@ -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]