Eval_Task
RGCN_prediction
-
neuralkg.eval_task.RGCN_prediction.RGCN_predict(batch, model, prediction='all')[source]
-
neuralkg.eval_task.RGCN_prediction.head_predict(batch, model)[source]
-
neuralkg.eval_task.RGCN_prediction.tail_predict(batch, model)[source]
-
neuralkg.eval_task.RGCN_prediction.calc_ranks(graph, ent, rel, norm, pos_triple, idx, label, mode, model)[source]