GraphClassificationTask

class gli.task.GraphClassificationTask(task_dict, pwd, device='cpu')

Bases: ClassificationTask

Graph classification task, alias.

__init__(task_dict, pwd, device='cpu')

Initialize num_classes.