更新 function.py

2024-10-17 09:03:59 +08:00
parent 0a556a6395
commit 05216b8c86
1 changed files with 2 additions and 2 deletions
@@ -82,8 +82,8 @@ def load_dataset(directory):
    """
    data = []  
    labels = []  
-      
-    for filename in os.listdir("D:\\two"):  
+    directory="D:\\two"
+    for filename in os.listdir(directory):  
        if filename.endswith(".txt"):  
            label = int(filename.split('_')[0])  # 假设文件名格式为 "label_xxx.txt"  
            with open(os.path.join(directory, filename), 'r') as f: