From 2b975c73452576bfcb6ad6d33e72f9e3df820245 Mon Sep 17 00:00:00 2001
From: Xionghaoran <719929980@qq.com>
Date: Wed, 23 Oct 2024 15:19:10 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/function.py b/function.py
index d5faba9..52376ab 100644
--- a/function.py
+++ b/function.py
@@ -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)