示例#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
示例#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
示例#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)
示例#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
示例#5
0
 def __init__(self, train):
     self.train = train
     super(AIDSData, self).__init__(get_train_str(train))
示例#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))
示例#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
示例#8
0
 def __init__(self, train):
     self.train = train
     super(IMDBMulti800Data, self).__init__(get_train_str(train))
示例#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))
示例#10
0
 def __init__(self, train):
     self.train = train
     super().__init__(get_train_str(train))
示例#11
0
 def __init__(self, train):
     self.train = train
     super(WebEasyData, self).__init__(get_train_str(train))
示例#12
0
 def __init__(self, train):
     self.train = train
     super(Reddit10kData, self).__init__(get_train_str(train))
示例#13
0
 def __init__(self, train):
     self.train = train
     super(ProjectData, self).__init__(get_train_str(train))
示例#14
0
 def __init__(self, train):
     self.train = train
     super(LINUX_IMDBMultiData, self).__init__(get_train_str(train))