diff --git a/Train2.py b/Train2.py
index 356f545c66c4ec3e9aec2eadb6940f3d186eb976..3b9ae86716f3a283de39f84a13267230574d9027 100644
--- a/Train2.py
+++ b/Train2.py
@@ -19,9 +19,9 @@ def load_data(DIR="subDataSet/EMB"):
     test_x = np.load("data_test.npy",allow_pickle=True)
     val_x = np.load("data_val.npy",allow_pickle=True)
     print(train_x[0].shape)
-    train_y = np.load("node2vec/src/graph/label_train.npy")
-    test_y = np.load("node2vec/src/graph/label_test.npy")
-    val_y= np.load("node2vec/src/graph/label_val.npy")
+    train_y = np.load("label_train.npy")
+    test_y = np.load("label_test.npy")
+    val_y= np.load("label_val.npy")
     print(val_y.shape)
 
     print(val_x[0])
@@ -94,4 +94,4 @@ for i in range(2900):
         tab = []
 
 end = time.time()
-"""
\ No newline at end of file
+"""