这是GNN-RAG:用于大型语言建模推理的图神经网络检索的代码。
目录结构如下:
|----gnn
文件夹包含不同KGQA GNN的实现。
你可以训练自己的GNN,或者跳过这个文件夹,直接使用我们计算的GNN输出(检索到的答案节点)(llm/results/gnn
)。
|----llm
文件夹包含基于RAG的KGQA与LLM的实现。
请在那里查看如何复现结果的详细信息。
结果:我们附加了表2的所有结果:请参见results/KGQA-GNN-RAG-RA
或results/KGQA-GNN-RAG
。你可以在predictions.jsonl中查看实际的LLM生成内容,以及检索到的KG信息("input"键)。