-
LanE:一款用于知识图谱补全的自然语言嵌入源代码,来自匿名ACL-IJCNLP提交
资源介绍
车道
纸自然语言嵌入知识图完成的源代码库。
设置数据集和依赖项
需要Python 3.7或更高版本。 Python 3.7.9已测试。
需要CUDA的PyTorch版本1.6.0或更高版本。 带有CUDA 11的PyTorch 1.6.0已经过测试。 请参阅来安装PyTorch。
python -m pip install -r requirements.txt
cd data
bash prepare_data.sh
cd ..
硬体需求
需要8个GPU。 运行BERT-Base或RoBERTa-Base需要12GB GPU内存。 运行BERT-Large或RoBERTa-Large所需的32GB GPU内存。 下载预训练的语言模型需要Internet连接。
复制结果的脚本
链接预测
bash FB15k-237_bert_base.sh
bash FB15k-237_bert_la