diff --git a/function.py b/function.py
index 15afaa0..fad802b 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=1):  # 设置默认值 k=1
+def classify(input_vector, dataSet, labels, k=2):  # 设置默认值 k=1
     """使用 KNN 算法对输入样本进行分类。"""
     # 计算输入向量与训练集之间的欧氏距离
     distances = np.linalg.norm(dataSet - input_vector, axis=1)