轻量级查询路由用于检索增强对话：基于置信度的TF-IDF和逻辑回归分类器

Key Points

• 目标是开发一个轻量级查询路由系统，用于检索增强对话（RAG）系统。
• 实现了一个五路查询路由器：NO_RAG、LOCAL_RAG、WEB_RAG、MEMORY_ONLY、ASK_CLARIFICATION。
• 利用TF-IDF和多项逻辑回归进行路由。
• 设定了0.45的置信度阈值作为备选。
• 进行了5折交叉验证（CV）以评估表现。
• 在测试数据上达到了93.6%的准确率。
• 获得了0.935的宏F1分数。
• 在平衡样本上达到了0.993的加权ROC AUC。