tasks.graph_classification

Module Contents

Classes

GraphClassification

Superiviced graph classification task.

Functions

node_degree_as_feature(data)	Set each node feature as one-hot encoding of degree
uniform_node_feature(data)	Set each node feature to the same

tasks.graph_classification.node_degree_as_feature(data)

Set each node feature as one-hot encoding of degree :param data: a list of class Data :return: a list of class Data

tasks.graph_classification.uniform_node_feature(data)

Set each node feature to the same

class tasks.graph_classification.GraphClassification(args)

Bases: tasks.BaseTask

Superiviced graph classification task.

static add_args(parser)

Add task-specific arguments to the parser.

train(self)

_train(self)

_train_step(self)

_test_step(self, split='val')

_kfold_train(self)

generate_data(self, dataset, args)