更新 function.py

2024-10-15 16:24:11 +08:00
parent 5264386762
commit dde1649b62
1 changed files with 1 additions and 1 deletions
@@ -45,7 +45,7 @@ import numpy as np
 import os
 from collections import Counter

-def classify(input_vector, dataSet, labels, k=3):  # 设置默认值 k=3
+def classify(input_vector, dataSet, labels, k=2):  # 设置默认值 k=2
    """使用 KNN 算法对输入样本进行分类。"""
    # 计算输入向量与训练集之间的欧氏距离
    distances = np.linalg.norm(dataSet - input_vector, axis=1)