更新 function.py

2024-10-16 10:35:34 +08:00
parent 2ef5aa0677
commit 096c45886e
1 changed files with 1 additions and 2 deletions
@@ -65,8 +65,7 @@ def classify0(inx, dataSet, labels, k_values=[3,8,9], _cache={"VI": None}):
    sortedDistIndices = np.argsort(distances)

    # 创建一个矩阵来保存每个k值的前k个样本的标签
-    max_k = max(k_values)
-    nearest_labels = labels[sortedDistIndices[:max_k]]  # 取最大的k个邻居的标签
+    nearest_labels = labels[sortedDistIndices[:-1]]  # 取最大的k个邻居的标签

    # 为每个k值计算预测结果，统计不同k值下出现最多的标签
    predictions = np.array([np.bincount(nearest_labels[:k]).argmax() for k in k_values])