轨道交通RAG应用数据检索方法、装置、设备、介质

本申请提供一种轨道交通RAG应用数据检索方法、装置、设备、介质，该方法包括：获取检索关键词；根据检索关键词，通过RAG系统进行数据检索，得到候选元数据和各候选元数据所嵌入向量的相似度；其中，候选元数据为轨道交通元数据中的一个或多个，且轨道交通元数据经过切分及向量嵌入后存入RAG系统中；通过DRMM模型得到各候选元数据的匹配分数；将各候选元数据的匹配分数与所嵌入向量的相似度之间的加权之和，确定为候选元数据的检索值；根据检索值得到检索得到的元数据，并反馈检索得到的元数据。本申请的方法通过RAG系统和DRMM模型实现了数据的多层级信息检索，可以有效提高轨道交通数据的检索效率和准确性。