diff --git a/function.py b/function.py
index 21bde09..88d774f 100644
--- a/function.py
+++ b/function.py
@@ -54,7 +54,7 @@ def classify0(inX, dataSet, labels, k=1):
     - int: 预测的分类标签。
     """
     dataSetSize=dataSet.shape[0]
-    diffMat=inX - dataset
+    diffMat=inX - dataSet
     sqDiffMat=diffMat**2
     sqDistances=sqDiffMat.sum(axis=1)
     distance=sqDistances**0.5