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)