def __init__(self, dataset, model, hidden_size, num_shuffle, seed,
**model_args):
self.data = create_node_classification_dataset(dataset).data
self.label_matrix = self.data.y
self.num_nodes, self.num_classes = self.data.y.shape
self.model = build_model(model, hidden_size, **model_args)
self.hidden_size = hidden_size
self.num_shuffle = num_shuffle
self.seed = seed
def __init__(self, dataset, model, hidden_size, num_shuffle, seed, **model_args):
assert model == "from_numpy_graph"
dataset = create_graph_classification_dataset(dataset)
self.num_nodes = dataset.graph_labels.shape[0]
self.num_classes = dataset.num_labels
self.label_matrix = np.zeros((self.num_nodes, self.num_classes), dtype=int)
self.labels = dataset.graph_labels
self.model = build_model(model, hidden_size, **model_args)
self.hidden_size = hidden_size
self.num_shuffle = num_shuffle
self.seed = seed
def __init__(self, dataset_1, dataset_2, model, hidden_size, **model_args):
self.data = SSDataset("data/panther", dataset_1, dataset_2).data
self.model = build_model(model, hidden_size, **model_args)
self.hidden_size = hidden_size