KGRelationPredictionTask

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

Bases: GLITask

Knowledge graph relation prediction task.

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

Rename num_relations to num_classes.