SMHSA:论文“注意关系”的源代码 Source code learning

SMHSA 论文的源代码“注意力作为关系:学习监督的多头自我注意进行关系提取,IJCAI2020。” 要求: python==3.6.9 torch==0.4.0 数据集和预训练的嵌入: 预先训练的手套840B嵌入:从下载 数据集:从下载。 如何运行: python dataProcess.py # For preprocessing dataset python loadPretrainEmbedding.py # For loading pre-trained embedding python main.py --mode train # For training python main.py --mode test --test_model ./test_model/modelFinal.model # For testing