def init(self):
self.graphs = []
datadir = '{}/LINUX/{}'.format(
get_data_path(), get_train_str(self.train))
self.graphs = iterate_get_graphs(datadir)
print('Loaded {} graphs from {}'.format(len(self.graphs), datadir))
self.graphs, self.glabels = add_glabel_to_each_graph(self.graphs, '', True)
assert (self.glabels is None) # fake graph labels
def get_proc_graphs(datadir, train):
if logging_enabled == True:
print("- Entered data::get_proc_graphs Global Method")
datadir = '{}\\{}\\{}'.format(
get_data_path(), datadir, get_train_str(train))
graphs = iterate_get_graphs(datadir)
print('info: Loaded {} graphs from {}'.format(len(graphs), datadir))
return graphs
def init(self):
self.graphs = []
datadir = '{}/{}/{}'.format(get_data_path(), self.get_folder_name(),
get_train_str(self.train))
self.graphs = iterate_get_graphs(datadir)
print('Loaded {} graphs from {}'.format(len(self.graphs), datadir))
if 'nef' in self.get_folder_name():
print('Removing edge features')
for g in self.graphs:
self._remove_valence(g)
def init(self):
self.graphs = []
datadir = '{}/{}/{}'.format(
get_data_path(), self.get_folder_name(), get_train_str(self.train))
self.graphs = iterate_get_graphs(datadir)
print('Loaded {} graphs from {}'.format(len(self.graphs), datadir))
if 'nef' in self.get_folder_name():
print('Removing edge features')
for g in self.graphs:
self._remove_valence(g)
self.graphs, self.glabels = add_glabel_to_each_graph(self.graphs, '', True)
assert (self.glabels is None) # fake graph labels
def __init__(self, train):
self.train = train
super(AIDSData, self).__init__(get_train_str(train))
def __init__(self, train):
if train:
self.num_graphs = SynData.train_num_graphs
else:
self.num_graphs = SynData.test_num_graphs
super(SynData, self).__init__(get_train_str(train))
def get_proc_graphs(datadir, train):
datadir = '{}/{}/{}'.format(
get_data_path(), datadir, get_train_str(train))
graphs = iterate_get_graphs(datadir)
print('Loaded {} graphs from {}'.format(len(graphs), datadir))
return graphs
def __init__(self, train):
self.train = train
super(IMDBMulti800Data, self).__init__(get_train_str(train))
def init(self):
self.graphs = []
datadir = '{}/IMDB1k{}/{}'.format(
get_data_path(), self._identity(), get_train_str(self.train))
self.graphs = iterate_get_graphs(datadir)
print('Loaded {} graphs from {}'.format(len(self.graphs), datadir))
def __init__(self, train):
self.train = train
super().__init__(get_train_str(train))
def __init__(self, train):
self.train = train
super(WebEasyData, self).__init__(get_train_str(train))
def __init__(self, train):
self.train = train
super(Reddit10kData, self).__init__(get_train_str(train))
def __init__(self, train):
self.train = train
super(ProjectData, self).__init__(get_train_str(train))
def __init__(self, train):
self.train = train
super(LINUX_IMDBMultiData, self).__init__(get_train_str(train))
