更新 function.py

function.py

@@ -60,7 +60,7 @@ def mahalanobis(inX, dataSet):
     
     return np.array(distances)
 
-def classify0(inX, dataSet, labels, k=20):
+def classify0(inX, dataSet, labels, k=9):
     """
     KNN 分类算法，使用马氏距离。
 
@@ -120,7 +120,7 @@ if __name__ == "__main__":
   
     # 对测试集中的样本进行预测
     for i in range(len(test_data)):
-        predicted_label = classify0(test_data[i], train_data, train_labels, k=20)
+        predicted_label = classify0(test_data[i], train_data, train_labels, k=9)
         print(f"样本 {i+1}: 预测标签: {predicted_label}, 实际标签: {test_labels[i]}")
         if predicted_label != test_labels[i]:
             a += 1