许可证：Apache-2.0
流水线标签：文本排序
库名称：lightning-ir
基础模型：

• google-bert/bert-base-uncased
  标签：
• 双编码器

Lightning IR SPLADE模型

本模型是基于Lightning IR框架微调的SPLADE[^1]模型。

与其他模型的性能对比，请参阅Lightning IR模型库。

复现步骤

1. 训练模型
   安装Lightning IR后，运行以下命令（使用微调配置文件）：

lightning-ir fit --config fine-tune.yaml  

2. 构建MS MARCO段落索引
   使用以下命令及索引配置文件：

lightning-ir index --config index.yaml  

3. 评估模型
   索引完成后，通过以下命令（配合搜索配置文件）在TREC Deep Learning 2019/2020数据集上评估：

lightning-ir search --config search.yaml  

[^1]: Formal等人, SPLADE：一阶段排序的稀疏词汇扩展模型