/KETM

基于知识增强的文本匹配模型

Primary LanguagePython

KETM

基于知识增强的文本匹配模型

运行步骤

  1. 运行add_knowledge生成知识文本，其中维基词典释义文件https://kaikki.org/dictionary/
  2. 运行generate文件，生成训练数据
  3. 将预训练语言模型下载到目录中
  4. 运行main.py 进行训练

注意事项

数据、模型的目录名记得按照你的目录进行修改

参考论文

《KETM:A Knowledge-Enhanced Text Matching method》