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')