Rankify RAG Python 工具包
Rankify 是一站式解决检索、重排和检索增强生成任务的综合性 Python 工具包,它提供了一个模块化和可扩展的框架,让使用者能够在检索管道中无缝进行实验和基准测试。
功能亮点
- 集成 40 个预检索数据集,覆盖问答、对话、实体链接等多领域
- 支持 7 种检索技术,包括 BM25、DPR、ANCE、BPR、ColBERT、BGE 和 Contriever,提供了各种检索策略的灵活性
- 24 种重排模型
- 实现了 24 种重排序模型和 41 个子方法,涵盖 pointwise、pairwise 和 listwise 重排序方法,以提高排名性能
- 提供预构建的 Wikipedia 和 MS MARCO 语料库,加速实验
评论