Python get_train_val_test_gcn Examples

Example #1

File: dataset.py Project: hmrg-grmh/DSE-MSU-DeepRobust

 def get_train_val_test(self):
     """Get training, validation, test splits according to self.setting (either 'nettack' or 'gcn').
     """
     if self.setting == 'nettack':
         return get_train_val_test(nnodes=self.adj.shape[0], val_size=0.1, test_size=0.8, stratify=self.labels, seed=self.seed)
     if self.setting == 'gcn':
         return get_train_val_test_gcn(self.labels, seed=self.seed)

Example #2

File: dataset.py Project: bradleybauer/intrinsic-dimensionality

    def get_train_val_test(self):

        if self.setting == 'nettack':
            return get_train_val_test(nnodes=self.adj.shape[0],
                                      val_size=0.1,
                                      test_size=0.8,
                                      stratify=self.labels,
                                      seed=self.seed)
        if self.setting == 'gcn':
            return get_train_val_test_gcn(self.labels, seed=self.seed)