更新 function.py

2024-10-17 09:46:39 +08:00
parent 515adc793a
commit 76cde940fd
1 changed files with 2 additions and 2 deletions
@@ -32,7 +32,7 @@ def img_to_vector(file_path):
    return vector

 # 使用欧式距离进行 kNN 分类
-def classify(input_vector, dataSet, labels, k=7):
+def classify(input_vector, dataSet, labels, k=3):
    # 计算所有训练样本与输入样本的欧式距离
    distances = np.linalg.norm(dataSet - input_vector, axis=1)
    # 获取距离最近的k个样本的标签
@@ -63,7 +63,7 @@ def main():

    for i in range(total_tests):
        test_sample = test_data[i]
-        prediction = classify(test_sample, training_data, training_labels, k=7)
+        prediction = classify(test_sample, training_data, training_labels, k=3)
        actual_label = test_labels[i]

        print(f"测试样本 {i+1}: 预测值 = {prediction}, 实际值 = {actual_label}")