diff --git a/function.py b/function.py index b935ee8..5ecc5ba 100644 --- a/function.py +++ b/function.py @@ -66,7 +66,7 @@ def classify0(inx, dataSet, labels, k_values=[3,8,9], _cache={"VI": None}): sortedDistIndices = np.argsort(distances) # 创建一个矩阵来保存每个k值的前k个样本的标签 - nearest_labels = labels[sortedDistIndices[:-1]] # 取最大的k个邻居的标签 + nearest_labels = labels[sortedDistIndices[:-1]] # 为每个k值计算预测结果,统计不同k值下出现最多的标签 predictions = np.array([np.bincount(nearest_labels[:k]).argmax() for k in k_values])