diff --git a/.gitignore b/.gitignore
index cb673bb..9de7b6a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -158,15 +158,26 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-.idea/
+#.idea/
 
+# 数据集文件屏蔽
 .STDEN/
-yuan
+
 exp/
 STDEN/
 models/gpt2/
 pre-trained/
-data/*
+
+# 数据集文件类型屏蔽
+*.csv
+*.npz
+*.npy
+*.pkl
+*.h5
+*.hdf5
+
+# 特定数据集目录屏蔽
 data/PEMS08/
 data/PEMSD8/
-
+test_semantic.npy
+test_spatial.npy
\ No newline at end of file