NodeClassificationTask

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

Bases: ClassificationTask

Node classification task, alias.

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

Initialize num_classes.