From 3aa4b46bfb654b7e481eed6d04477e2a66d1641a Mon Sep 17 00:00:00 2001 From: linxing719 Date: Thu, 26 Aug 2021 17:05:27 +0800 Subject: [PATCH] train.py line 60 add / --- src/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/train.py b/src/train.py index d6aa866..364fddd 100644 --- a/src/train.py +++ b/src/train.py @@ -57,7 +57,7 @@ def train(params): print("Loading data") ORIG_DATA_DIR = os.path.join(ROOT, 'mag240m_kddcup2021/processed') - node_label_all = np.load(f'{ROOT}mag240m_kddcup2021/processed/paper/node_label.npy') + node_label_all = np.load(f'{ROOT}/mag240m_kddcup2021/processed/paper/node_label.npy') year = np.load(os.path.join(ORIG_DATA_DIR, 'paper/node_year.npy')) bert_features = np.load(os.path.join(ORIG_DATA_DIR, 'paper/node_feat.npy'), mmap_mode='r') edges_paper_cite_paper_sorted_by_first = np.load(os.path.join(ORIG_DATA_DIR, 'paper___cites___paper/edge_index.npy'), mmap_mode='r') @@ -193,4 +193,4 @@ def train(params): if __name__ == "__main__": parser = get_parser() params = parser.parse_args() - train(params) \ No newline at end of file + train(params)