更新 function.py

function.py

@@ -8,27 +8,6 @@ import os
 - 函数输入: 待分类的样本 input_vector, 训练数据集 dataSet, 训练集对应的标签 labels。
 - 函数返回: 预测的分类标签。
 
-注意事项:
-- 在 function.py 中实现 kNN 分类算法。
-- 可以自行导入其他库，需要在 requirements.txt 添加依赖，若未在 requirements.txt 中添加使用到的依赖，则无法自动评测，成绩为零。
-- 自动评测脚本仅会加载 classify 和 load_data 函数，请勿修改函数名。
-- 训练集和测试集需手动下载解压，并放入当前目录下。
-
-测评环境:
-- CPU: Intel Xeon(R) CPU E3-1231 v3 @ 3.40GHz
-- OS: Ubuntu 22.04.4 LTS
-- Python: 3.10.12
-- numpy: 2.1.1
-"""
-
-
-"""
-请在这里填写学号和姓名，不填写即成绩无效。
-"""
-id = "202210423213"
-name = "熊浩燃"
-
-
 """
 参数说明:
 - input_vector (numpy array): 输入待分类的样本。
@@ -43,44 +22,25 @@ name = "熊浩燃"
 # 这里写自己的代码实现
 def classify(input_vector, dataSet, labels, k=1):
     """
-    KNN 分类算法，使用加权多数表决法。
+    KNN 分类算法。
 
     Args:
     - input_vector (numpy array): 输入待分类的样本。
     - dataSet (numpy array): 训练数据集。
     - labels (list or numpy array): 训练集对应的标签。
-    - k (int): kNN 算法参数。
+    - k (int): kNN 算法参数，请**自行设置默认值**，评测时仅会使用其默认值，不修改默认值将**无法通过**评测，**成绩为零**。
 
     Returns:
     - int: 预测的分类标签。
     """
     # 计算欧氏距离
     distances = np.linalg.norm(dataSet - input_vector, axis=1)
-
     # 获取距离最近的 k 个样本的索引
     nearest_neighbors_indices = distances.argsort()[:k]
-
     # 获取这些样本的标签
     nearest_labels = np.array(labels)[nearest_neighbors_indices]
-    
-    # 获取这些样本的距离
-    nearest_distances = distances[nearest_neighbors_indices]
-    
-    # 防止除以零的情况，添加一个小常数epsilon
-    epsilon = 1e-5
-    weights = 1 / (nearest_distances + epsilon)  # 距离越小，权重越大
-    
-    # 对每个标签进行加权投票
-    label_weights = {}
-    for i in range(k):
-        label = nearest_labels[i]
-        if label not in label_weights:
-            label_weights[label] = 0
-        label_weights[label] += weights[i]
-    
-    # 返回加权得分最高的标签
-    predicted_label = max(label_weights, key=label_weights.get)
-
+    # 统计出现次数最多的标签并返回
+    predicted_label = np.bincount(nearest_labels).argmax()
     return predicted_label
 
 # 这里也写自己的代码实现