基于BERT的两阶段民间故事检索方法

摘要

一种基于BERT模型的两阶段民间故事检索方法，由收集民间故事、民间故事数据预处理、构建民间故事数据集、一阶段构建向量搜索引擎、筛选候选民间故事集合、训练BERT模型、二阶段确定相关度、展示检索结果步骤组成。采用本发明与现有的传统检索方法进行了对比试验，实验结果表明，本发明可以更好地了解民间故事的上下文信息，更好地将查询请求与民间故事结合起来，在提升了检索准确率的同时还加快了检索速度。本发明具有检索结果准确、检索速度快等特点，可以在海量的民间故事中准确找到用户想要了解的民间故事。