更新 function.py

function.py

@@ -83,6 +83,15 @@ def load_dataset(directory):
     data = []  
     labels = []  
     directory="D:\\two"
+    try:  
+        files = os.listdir(directory)  
+        print("Files in the directory:", files)  
+    except FileNotFoundError:  
+        print(f"The directory '{directory}' does not exist.")  
+    except PermissionError:  
+        print(f"You do not have permission to access the directory '{directory}'.")  
+    except Exception as e:  
+        print(f"An unexpected error occurred: {e}")
     for filename in os.listdir(directory):  
         if filename.endswith(".txt"):  
             label = int(filename.split('_')[0])  # 假设文件名格式为 "label_xxx.txt"