From 9deefa48172b88983392c1cbd0a2956dce91fc02 Mon Sep 17 00:00:00 2001
From: wujunjie <297081150@qq.com>
Date: Thu, 26 Sep 2024 10:57:47 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/function.py b/function.py
index a694402..86aba4b 100644
--- a/function.py
+++ b/function.py
@@ -50,11 +50,16 @@ def classify0(inX, dataSet, labels,k=4):
     Returns:
     - int: 预测的分类标签。
     """
-    dataSetSize = dataSet.shape[0]  
-    diffMat = np.tile(inX, (dataSetSize, 1)) - dataSet  
+    if dataSet.shape[0] == 0:  
+        raise ValueError("数据集为空")  
+      
+    # 确保 k 不大于数据集的大小  
+    k = min(k, dataSet.shape[0])  
+      
+    # 计算测试样本与每个训练样本之间的距离  
+    diffMat = np.tile(inX, (dataSet.shape[0], 1)) - dataSet  
     sqDiffMat = diffMat ** 2  
-    sqDistances = sqDiffMat.sum(axis=1)  
-    distances = sqDistances ** 0.5  
+    distances = np.sqrt(sqDiffMat.sum(axis=1))  
       
     # 对距离进行排序，并获取索引  
     sortedDistIndicies = distances.argsort()