From 252f0e84ba45c6e9f7d34605d9247a601887dccf Mon Sep 17 00:00:00 2001
From: wzy <675163717@qq.com>
Date: Mon, 14 Oct 2024 20:56:42 +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 | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/function.py b/function.py
index b6402bf..6141d77 100644
--- a/function.py
+++ b/function.py
@@ -1,6 +1,8 @@
 # function.py
 import numpy as np
 import os
+from scipy.spatial.distance import euclidean
+from collections import Counter
 
 """
 实验说明:
@@ -54,7 +56,7 @@ def classify(input_vector, dataSet, labels, k=3):
     Returns:
     - int: 预测的分类标签。
     """
-    dists = [distance.euclidean(input_vector, data) for data in dataSet]
+    dists = [euclidean(input_vector, data) for data in dataSet]
     k_indices = np.argsort(dists)[:k]
     k_nearest_labels = [labels[i] for i in k_indices]