Rankify RAG Python 工具包

联合创作 · 2025-02-18 07:50

Rankify 是一站式解决检索、重排和检索增强生成任务的综合性 Python 工具包,它提供了一个模块化和可扩展的框架,让使用者能够在检索管道中无缝进行实验和基准测试。

功能亮点

  • 集成 40 个预检索数据集,覆盖问答、对话、实体链接等多领域
  • 支持 7 种检索技术,包括 BM25、DPR、ANCE、BPR、ColBERT、BGE 和 Contriever,提供了各种检索策略的灵活性
  • 24 种重排模型
  • 实现了 24 种重排序模型和 41 个子方法,涵盖 pointwise、pairwise 和 listwise 重排序方法,以提高排名性能
  • 提供预构建的 Wikipedia 和 MS MARCO 语料库,加速实验

浏览 4
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报