LightRAG

探索 LightRAG 实现 本节详细介绍如何复现 LightRAG 的结果。

LightRAG 索引

LightRAG 是我们实验的一个快照,其参数、函数和提示都经过微调,以返回统计数据并使用统一的提示。要开始使用,您首先需要创建一个新环境并安装 LightRAG 依赖项。

conad create -n lightrag python=3.10
conda activate lightrag
cd LightRAG
pip install -e .

与其他 RAG 实现类似,您需要创建一个名为 main_folder 的主工作目录,并在其中放置一个 input 文件夹来存储您的语料库文件。

main_folder/
├── input/
│   ├── file1.md
│   ├── file2.txt
│   ├── file3.docx
│   └── ...

然后运行

python -m Light_index -f path/to/main_folder

回答和评估

首先,根据基准测试格式准备您的测试问题。您需要创建一个包含问题及其对应答案键的测试集 parquet 文件。准备好后,您可以使用以下命令运行评估:

python -m /eval/eval_light -f path/to/main_folder -q path/to/question_parquet

Last modified April 5, 2025: update reproduce (f23a25c)