From f46a3213f8a401d4525a5d4c6262659cff29969f Mon Sep 17 00:00:00 2001 From: 202210423203 <2579478480@qq.com> Date: Wed, 16 Oct 2024 22:34:36 +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 | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/function.py b/function.py index e5f3050..bcb0277 100644 --- a/function.py +++ b/function.py @@ -23,7 +23,7 @@ import os """ 请在这里填写学号和姓名,不填写即成绩无效。 """ -id = "20221042303" +id = "202210423203" name = "赵昊龙" @@ -55,7 +55,8 @@ def classify(input_vector, dataSet, labels, k=3): # 计算输入向量与数据集中每个向量的欧式距离 distances = [] for i in range(dataSet.shape[0]): - dist = np.sqrt(np.sum((input_vector - dataSet[i]) ** 2)) + diff = (input_vector - dataSet[i]).astype(float) # 转换为浮点数 + dist = np.sqrt(np.sum(diff ** 2)) distances.append((dist, labels[i])) # 根据距离进行排序,取前k个最近邻 @@ -154,4 +155,4 @@ if __name__ == "__main__": train_labels, ) - print(f"预测的标签: {predicted_label}") + print(f"预测的标签: {predicted_label}") \ No newline at end of file