This is the multi-page printable view of this section. Click here to print.

Return to the regular view of this page.

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