嵌入KGQA
这是我们的ACL 2020论文的代码,
更新:用于关系匹配的代码已添加。请参阅以获取有关如何使用它的详细信息。
指示
为了运行代码,请首先从下载data.zip和pretrained_model.zip。将这些文件解压缩到主目录中。
MetaQA
转到目录./KGQA/LSTM。以下是运行质量检查培训代码的示例命令
python3 main.py --mode train --relation_dim 200 --hidden_dim 256 \
--gpu 2 --freeze 0 --batch_size 128 --validate_every 5 --hops 2 --lr 0.0005 --entdrop 0.1 --reldrop 0.2 --scoredrop 0.2 \
--decay 1.0 --model ComplEx --patience 5 -
2022-05-21 11:48:52
3.02MB
Python
1