diff --git a/function.py b/function.py
index dfe47c2..aac2ae2 100644
--- a/function.py
+++ b/function.py
@@ -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])