更新 function.py

2024-10-23 15:19:10 +08:00
parent 21ba1329b8
commit 2b975c7345
1 changed files with 5 additions and 0 deletions
@@ -52,6 +52,11 @@ def classify(input_vector, dataSet, labels, k=1):
    """
    # 计算数据集的协方差矩阵
    covariance_matrix = np.cov(dataSet, rowvar=False)
+    
+    # 添加正则化项，防止协方差矩阵奇异
+    regularization_term = 1e-5 * np.eye(covariance_matrix.shape[0])
+    covariance_matrix += regularization_term
+    
    # 计算协方差矩阵的逆
    inv_covariance_matrix = np.linalg.inv(covariance_matrix)