From 1a4d85391987f6342e42837a7dcc0a2cf8f88445 Mon Sep 17 00:00:00 2001 From: Xionghaoran <719929980@qq.com> Date: Tue, 22 Oct 2024 20:05:44 +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 | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/function.py b/function.py index bf7ac7c..013c8be 100644 --- a/function.py +++ b/function.py @@ -98,9 +98,10 @@ def img_to_vector(filename): if __name__ == "__main__": - # 测试代码 + # 加载训练数据 train_data, train_labels = load_dataset("trainingDigits") + # 测试数据 test_data = ( "00000000000000110000000000000000" "00000000000011111000000000000000" @@ -138,6 +139,6 @@ if __name__ == "__main__": test_vector = np.array([int(char) for char in test_data]).reshape(1, 1024) - # 使用默认的欧氏距离和不加权的投票机制 - predicted_label = classify(test_vector, train_data, train_labels) - print(f"预测的标签(普通分类): {predicted_label}") + # 使用曼哈顿距离进行分类 + predicted_label = classify(test_vector, train_data, train_labels, distance_metric='manhattan') + print(f"预测的标签(曼哈顿距离): {predicted_label}")