main
parent
c2c081e0df
commit
df9704e595
3 changed files with 33 additions and 5 deletions
@ -0,0 +1,27 @@ |
||||
import pickle |
||||
|
||||
import numpy as np |
||||
import scipy.sparse as sp |
||||
|
||||
|
||||
def load_data_1(dataset): |
||||
adj = np.loadtxt('./data/partitioned_data/{}/orig/e_adj_orig.txt'.format(dataset), dtype=int) |
||||
adj = sp.csr_matrix(adj) |
||||
|
||||
features = pickle.load(open("data/partitioned_data/e/feature/e_feature.pkl",'rb')) |
||||
|
||||
y_test = 0 |
||||
|
||||
tx = 0 |
||||
|
||||
ty = 0 |
||||
|
||||
test_mask = 0 |
||||
|
||||
labels = 0 |
||||
|
||||
return adj, features, y_test, tx, ty, test_mask, labels |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
load_data_1('e') |
Loading…
Reference in new issue