更新 function.py

2024-10-16 19:58:54 +08:00
parent a87e27842f
commit adeafa332a
1 changed files with 2 additions and 2 deletions
@@ -1,6 +1,6 @@
 # function.py
 import numpy as np
-from collections import defaultdict
+

 """
 挑战说明:
@@ -78,7 +78,7 @@ def classify0(inX, dataSet, labels, k=3):
    distances = np.array([mahalanobis_distance(inX, x, cov_inv, mean) for x in dataSet])
    
    # 获取距离最近的 k 个样本的索引
-    sortedDistIndices = argpartition(distances, k)[:k]
+    sortedDistIndices = np.argpartition(distances, k)[:k]
    
    # 获取这些样本对应的标签
    classCount = {}