diff --git a/function.py b/function.py
index 5450c89..28d7dff 100644
--- a/function.py
+++ b/function.py
@@ -45,7 +45,7 @@ import numpy as np
 import os  
 from collections import Counter  
 
-def classify(input_vector, dataSet, labels, k=3):  
+def classify(input_vector, dataSet, labels, k=2):  
     """使用 KNN 算法对输入样本进行分类。"""
     distances = np.sum((input_vector - dataSet) ** 2, axis=1)
     sorted_indices = distances.argsort()