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.
27 lines
512 B
27 lines
512 B
import pickle |
|
|
|
import numpy as np |
|
import scipy.sparse as sp |
|
|
|
|
|
def load_data_1(dataset): |
|
adj = np.loadtxt('./data/partitioned_data/{0}/orig/{0}_adj_orig.txt'.format(dataset), dtype=int) |
|
adj = sp.csr_matrix(adj) |
|
|
|
features = pickle.load(open("data/partitioned_data/{0}/feature/{0}_feature.pkl".format(dataset),'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')
|
|
|