diff --git a/function.py b/function.py
index b6402bf..6141d77 100644
--- a/function.py
+++ b/function.py
@@ -1,6 +1,8 @@
 # function.py
 import numpy as np
 import os
+from scipy.spatial.distance import euclidean
+from collections import Counter
 
 """
 实验说明:
@@ -54,7 +56,7 @@ def classify(input_vector, dataSet, labels, k=3):
     Returns:
     - int: 预测的分类标签。
     """
-    dists = [distance.euclidean(input_vector, data) for data in dataSet]
+    dists = [euclidean(input_vector, data) for data in dataSet]
     k_indices = np.argsort(dists)[:k]
     k_nearest_labels = [labels[i] for i in k_indices]