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