更新 function.py

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}")