yao8839836 / kg-llm

Exploring large language models for knowledge graph completion
96 stars 7 forks source link

Tail Entity Prediction #7

Open JiyaoWei opened 5 months ago

JiyaoWei commented 5 months ago

您好,可以提供一份表4中的尾实体预测的代码吗?

yao8839836 commented 5 months ago

@JiyaoWei 您好。代码跟Readme中提到py文件差不多,主要区别在于训练、推理的文件,下面是两个测试文件的例子,既包含预测头实体、也包含预测尾实体。

test_instructions_glm_merge.json

https://github.com/yao8839836/kg-llm/blob/main/data/WN18RR/test_instructions_llama_merge.csv