Ejemplo n.º 1
0
 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
Ejemplo n.º 2
0
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
Ejemplo n.º 3
0
 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)
Ejemplo n.º 4
0
 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
Ejemplo n.º 5
0
 def __init__(self, train):
     self.train = train
     super(AIDSData, self).__init__(get_train_str(train))
Ejemplo n.º 6
0
 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))
Ejemplo n.º 7
0
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
Ejemplo n.º 8
0
 def __init__(self, train):
     self.train = train
     super(IMDBMulti800Data, self).__init__(get_train_str(train))
Ejemplo n.º 9
0
 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))
Ejemplo n.º 10
0
 def __init__(self, train):
     self.train = train
     super().__init__(get_train_str(train))
Ejemplo n.º 11
0
 def __init__(self, train):
     self.train = train
     super(WebEasyData, self).__init__(get_train_str(train))
Ejemplo n.º 12
0
 def __init__(self, train):
     self.train = train
     super(Reddit10kData, self).__init__(get_train_str(train))
Ejemplo n.º 13
0
 def __init__(self, train):
     self.train = train
     super(ProjectData, self).__init__(get_train_str(train))
Ejemplo n.º 14
0
 def __init__(self, train):
     self.train = train
     super(LINUX_IMDBMultiData, self).__init__(get_train_str(train))