更新 function.py

2024-10-20 14:00:56 +08:00
parent 5ef77a23dc
commit c9cc3392e1
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):  
+def classify(input_vector, dataSet, labels, k=2):  
    """使用 KNN 算法对输入样本进行分类。"""
    distances = np.sum((input_vector - dataSet) ** 2, axis=1)
    sorted_indices = distances.argsort()