Eval_Task

CompGCN_prediction

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]

XTransE_prediction