From aeaa2a4715bbd775c43bdfde828020672bed492a Mon Sep 17 00:00:00 2001
From: Xionghaoran <719929980@qq.com>
Date: Mon, 14 Oct 2024 21:39:13 +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 | 15 ++-------------
 1 file changed, 2 insertions(+), 13 deletions(-)

diff --git a/function.py b/function.py
index 0935b82..d54c270 100644
--- a/function.py
+++ b/function.py
@@ -88,24 +88,13 @@ def load_data(file_path):
     labels = data[:, -1].astype(int)
     return features, labels
 
-if __name__ == "__main__":
-    train_file = "datingTestSet2.tsv"
-    train_data, train_labels = load_data(train_file)
-    
-    # 预测结果
-    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels, k=3))
 
-    
 
 
 if __name__ == "__main__":
     # 以下代码可用于测试，可自行随意修改
-    train_file = "datingTrainSet2.tsv"
+    train_file = "datingTestSet2.tsv"
     train_data, train_labels = load_data(train_file)
     
-    # 找到最佳 k 值
-    best_k = find_best_k(train_data, train_labels)
-    print(f"Best k: {best_k}")
-    
     # 预测结果
-    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels, best_k))
\ No newline at end of file
+    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels, k=3))
\ No newline at end of file