KBQA_zh
基于bert的KBQA,包含joint和pipeline两种模式
Introduction
本项目是一个基于知识三元组的简单的问答系统,分为joint和pipeline两种模式。joint模式为基于bert做意图识别和命名体识别的联合学习训练得到的模型,pipeline模式为单独训练命名实体识别和QA相似度模型。
Environment
tensorflow 1.15
python 3.6
pymysql 0.10.1
grpcio 1.32.0
grpcio-tools 1.32.0
How to run
首先需要下载bert的中文预训练模型放在chinese_L-12_H-768_A-12文件夹下,
然后,以pipeline模式为例,执行如下命令,训练命名实体识别模型和问答模型:
1. cd qa_pipeline
2. sh run_ner.sh
3. sh run_
2023-02-18 15:35:51
4.68MB
Python
1