更新 function.py

2024-10-15 19:16:24 +08:00
parent 12a6699575
commit 0d1fe66744
1 changed files with 1 additions and 1 deletions
@@ -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):