diff --git a/function.py b/function.py
index edc5b20..541182a 100644
--- a/function.py
+++ b/function.py
@@ -61,7 +61,7 @@ def classify0(inX, dataSet, labels, k=8, _cache={"VI": None, "dataSet": None}):
     diff = dataSet - inX
 
     # 使用矩阵运算计算马氏距离
-    distances = np.einsum("ij,jk,ik->i", diff, _cache["VI"], diff)
+    distances = np.sum(diff @ _cache["VI"] * diff, axis=1)
 
     # 对距离排序，获取前 k 个最近的样本索引
     sortedDistIndices = np.argpartition(distances, 8)[:8]