search

WEIYanbin1999 / KICGPT

[EMNLP 2023] KICGPT: Large Language Model with Knowledge in Context for Knowledge Graph Completion
MIT License
33 stars 3 forks source link

How can I generate retriever_candidate #1

Closed qiaoqiao228 closed 7 months ago

qiaoqiao228 commented 7 months ago

I plan to use my own built dataset on the program, realizing that the program requires "retriever_candidate_tail. txt" and "retriever_candidate_head. txt". I noticed that the dataset you provided includes the above files. How can I generate a retriever on my own dataset. Thank you for your help.

WEIYanbin1999 commented 7 months ago

To get retriever results, you should first choose a classic/advanced link prediction retriever. For example, in our experiment, the retriever is set as RotatE. Then you should use this retriever(RotatE) to run the link prediction, and record its top m prediction results in "retriever_candidate_tail. txt" and "retriever_candidate_head. txt".