You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
489 B
28 lines
489 B
2 years ago
|
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')
|