From 0a556a63955d3fe7a73b4e4fcd92b1e0a7ae0b74 Mon Sep 17 00:00:00 2001 From: zhe <2904040094@qq.com> Date: Thu, 17 Oct 2024 09:01:43 +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 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/function.py b/function.py index e17d7ae..6b900f5 100644 --- a/function.py +++ b/function.py @@ -83,7 +83,7 @@ def load_dataset(directory): data = [] labels = [] - for filename in os.listdir(directory): + for filename in os.listdir("D:\\two"): if filename.endswith(".txt"): label = int(filename.split('_')[0]) # 假设文件名格式为 "label_xxx.txt" with open(os.path.join(directory, filename), 'r') as f: