Colbert

🚀 Lightning IR ColBERT

Lightning IR ColBERT是一個經過微調的ColBERT模型，使用Lightning IR進行訓練。該模型可用於文本排序任務，能為相關應用提供高效準確的文本排序能力。

🚀 快速開始

模型復現

若要復現模型訓練，需先安裝Lightning IR，然後使用fine-tune.yaml配置文件運行以下命令：

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

索引構建

若要為MS~MARCO段落構建索引，可使用index.yaml配置文件並運行以下命令：

lightning-ir index --config index.yaml

模型評估

在完成索引構建後，若要在TREC Deep Learning 2019和2020上評估模型，可使用search.yaml配置文件並運行以下命令：

lightning-ir search --config search.yaml

📚 詳細文檔

可參考Lightning IR Model Zoo，以瞭解該模型與其他模型的對比情況。

📄 許可證

本項目採用Apache-2.0許可證。

引用信息

該模型基於ColBERT方法，相關論文為：Khattab和Zaharia的ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERT。