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