更新 function.py

2024-11-21 09:21:10 +08:00
parent c97731bf9e
commit 32b108630f
1 changed files with 1 additions and 1 deletions
@@ -50,7 +50,7 @@ def classify0(inX, dataSet, labels, k=3):
    inX = (inX - np.mean(dataSet, axis=0)) / np.std(dataSet, axis=0)
    
    # 计算距离
-    distances = [distance.euclidean(inX, x) for x in dataSet]
+    distances = np.sqrt(np.sum((dataSet - inX) ** 2, axis=1))
    
    # 获取k个最近邻的索引
    k_indices = np.argsort(distances)[:k]