def get_mnist_data_gcn(perm):
N, train_data, train_labels, test_data, test_labels = load_mnist()
train_data = coarsening.perm_data(train_data, perm)
test_data = coarsening.perm_data(test_data, perm)
del perm
return train_data, test_data, train_labels, test_labels
def get_MNIST_Data_Autograd(perm):
N, train_data, train_labels, test_data, test_labels = load_mnist()
idx_train = range(0, 20*512)
idx_test = range(0, 10*512)
train_data = train_data[idx_train]
train_labels = train_labels[idx_train]
test_data = test_data[idx_test]
test_labels = test_labels[idx_test]
train_data = coarsening.perm_data(train_data, perm)
test_data = coarsening.perm_data(test_data, perm)
del perm
return train_data, test_data, train_labels, test_labels
def get_mnist_data_gcn(perm):
N, train_data, train_labels, test_data, test_labels = load_mnist()
# sz_batch = 512
# n_batch_train = 40
# n_batch_test = 5
#
# idx_train = range(0, n_batch_train*sz_batch)
# idx_test = range(0, n_batch_test*sz_batch)
#
# train_data = train_data[idx_train]
# train_labels = train_labels[idx_train]
# test_data = test_data[idx_test]
# test_labels = test_labels[idx_test]
train_data = coarsening.perm_data(train_data, perm)
test_data = coarsening.perm_data(test_data, perm)
del perm
return train_data, test_data, train_labels, test_labels