diff --git a/function.py b/function.py
index 9e9362d..1c00292 100644
--- a/function.py
+++ b/function.py
@@ -44,7 +44,7 @@ def classify(input_vector, dataSet, labels, k=2):
     dataSet_size = dataSet.shape[0]
     # 计算马氏距离
     cov_matrix = np.cov(dataSet.T)
-    inv_cov_matrix = np.linalg.inv(cov_matrix)
+    inv_cov_matrix = np.linalg.pinv(cov_matrix)
     diff_mat = np.tile(input_vector, (dataSet_size, 1)) - dataSet
     distances = []
     for i in range(dataSet_size):