From 0d1754a486677672c55390dcf099be33f0138a6a Mon Sep 17 00:00:00 2001
From: heart20003 <1789588961@qq.com>
Date: Thu, 17 Oct 2024 16:16:42 +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 | 36 ++++++++++++++----------------------
 1 file changed, 14 insertions(+), 22 deletions(-)

diff --git a/function.py b/function.py
index d0331e4..8ed6546 100644
--- a/function.py
+++ b/function.py
@@ -39,38 +39,30 @@ name = "陈垣豪"
 - int: 预测的分类标签。
 """
 
-def classify0(inX, dataSet, labels, k=3):
-    # Normalize the dataset
-    dataSet_mean = np.mean(dataSet, axis=0)
-    dataSet_std = np.std(dataSet, axis=0)
-    normalized_dataSet = (dataSet - dataSet_mean) / dataSet_std
-    normalized_inX = (inX - dataSet_mean) / dataSet_std
-    
+def classify0(inX, 数据集， 标签， k=3):
     # 计算协方差矩阵并求其逆矩阵
-    cov_matrix = np.cov(normalized_dataSet, rowvar=False)
+    cov_matrix = np.cov(数据集,rowvar=False)
     inv_cov_matrix = np.linalg.inv(cov_matrix)
-    
+
     # 计算马氏距离
-    diff = normalized_dataSet - normalized_inX
-    distances = np.einsum('ij,jk->i', diff, inv_cov_matrix @ diff.T)  # 使用 einsum 简化计算
-    
+    diff = dataSet - inX
+    distances = np.sum(diff @ inv_cov_matrix * diff, axis=1)
+
     # 获取距离最小的 k 个样本的索引
     sorted_indices = np.argsort(distances)[:k] 
-    
+
     # 获取 k 个最近邻的标签
     nearest_labels = [labels[i] for i in sorted_indices]
-    
+
     # 统计 k 个最近邻中每个类别的出现次数
     label_count = {}
-    weights = 1 / (distances[sorted_indices] + 1e-10)  # Avoid division by zero
-    
-    for i, label in enumerate(nearest_labels):
+    for label in nearest_labels:
         if label in label_count:
-            label_count[label] += weights[i]
+            label_count[label] += 1
         else:
-            label_count[label] = weights[i]
-    
+            label_count[label] = 1
+
     # 找到出现次数最多的标签
     predicted_label = max(label_count, key=label_count.get)
-    
-    return predicted_label
+
+    return predicted_label提高准确率