更新 function.py

function.py

@@ -1,36 +1,9 @@
-# function.py
+#function.py
 import numpy as np
 import os
-
-"""
-实验说明:
-- 自行编写实现数据加载与 kNN 的手写数字分类算法。
-- 函数输入: 待分类的样本 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 = "202210423115"
-name = "李煜锋"
-# function.py
-import numpy as np
-import os
-
+from collections import Counter
+id="202210423115"
+name="李煜锋"
 def img_to_vector(filename):
     """
     将二值图像转换为向量。
@@ -60,6 +33,7 @@ def load_dataset(directory):
     feature_list = []
     label_list = []
     
+    print(f"正在加载目录: {directory}")
     # 遍历目录，加载数据
     for filename in os.listdir(directory):
         if filename.endswith('.txt'):
@@ -78,7 +52,6 @@ def load_dataset(directory):
     labels = np.array(label_list)
     
     return features, labels
-from collections import Counter
 
 def knn_classify(input_vector, dataSet, labels, k=3):
     """
@@ -106,14 +79,29 @@ def knn_classify(input_vector, dataSet, labels, k=3):
     label_count = Counter(top_k_labels)
     
     return label_count.most_common(1)[0][0]
-# 加载训练数据
-train_data, train_labels = load_dataset('C:\\Users\\Administrator\\Desktop\\trainingDigits')
 
-# 加载测试数据
-test_data, test_labels = load_dataset('C:\\Users\\Administrator\\Desktop\\testDigits')
-
-# 选择一个测试样本，进行分类
-for i in range(len(test_data)):
-    predicted_label = knn_classify(test_data[i], train_data, train_labels, k=3)
-    true_label = test_labels[i]
-    print(f"预测标签: {predicted_label}, 真实标签: {true_label}")
+# 主程序
+if __name__ == '__main__':
+    # 使用绝对路径替换为你数据集的路径
+    train_directory = r'C:\Users\Administrator\Desktop\trainingDigits'
+    test_directory = r'C:\Users\Administrator\Desktop\testDigits'
+    
+    # 加载训练数据
+    train_data, train_labels = load_dataset(train_directory)
+    
+    # 加载测试数据
+    test_data, test_labels = load_dataset(test_directory)
+    
+    # 测试 kNN 分类器
+    correct_predictions = 0
+    
+    for i in range(len(test_data)):
+        predicted_label = knn_classify(test_data[i], train_data, train_labels, k=3)
+        true_label = test_labels[i]
+        print(f"预测标签: {predicted_label}, 真实标签: {true_label}")
+        
+        if predicted_label == true_label:
+            correct_predictions += 1
+            
+    accuracy = correct_predictions / len(test_data) * 100
+    print(f"分类准确率: {accuracy:.2f}%")