KGEntityPredictionTask

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

Bases: GLITask

Knowledge graph entity prediction task.

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

Initialize KGEntityPredictionTask.