(dataset_dir=datadir)
| 118 | |
| 119 | |
| 120 | def loadRedditFromNPZ(dataset_dir=datadir): |
| 121 | adj = sp.load_npz(dataset_dir+"reddit_adj.npz") |
| 122 | data = np.load(dataset_dir +"reddit.npz") |
| 123 | |
| 124 | return adj, data['feats'], data['y_train'], data['y_val'], data['y_test'], data['train_index'], data['val_index'], data['test_index'] |
| 125 | |
| 126 | |
| 127 | def load_reddit_data(normalization="AugNormAdj", porting_to_torch=True, data_path=datadir): |