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)
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)