Skip to content

RUCAIBox/RioRAG

Β 
Β 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

5 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

RioRAG

RioRAG is a reinforcement learning-based framework for improving long-form Retrieval-Augmented Generation (RAG).

πŸš€ Quick Start

πŸ”§ Installation

Please install the following dependencies:

  • verl==0.2.0
  • vllm==0.8.2
  • ragchecker
  • refchecker
  • demjson

πŸ—‚οΈ Data Preparation

cd train/verl/verl
python data_preprocess/checklist_rl.py

πŸ‹οΈβ€β™€οΈ Training

cd train/verl
bash scripts/run_checklist.sh

πŸ§ͺ Evaluation

Please refer to the code in the eval/ directory for evaluation.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 78.5%
  • Jupyter Notebook 21.5%