更新 function.py

2024-10-17 09:01:43 +08:00
parent a5e99ef366
commit 0a556a6395
1 changed files with 1 additions and 1 deletions
@@ -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: