diff --git a/README.md b/README.md index b1150ed3..e97f84ea 100644 --- a/README.md +++ b/README.md @@ -133,7 +133,7 @@ pip install dingo-python # With HHEM hallucination detection model (requires transformers + torch) pip install "dingo-python[hhem]" -# With retrieval benchmark evaluation (MTEB + pytrec_eval) +# With retrieval benchmark evaluation (MTEB + pytrec-eval-terrier) pip install "dingo-python[retrieval]" # With all features (HHEM + Agent + Retrieval) diff --git a/README_ja.md b/README_ja.md index 38414da3..948a16f6 100644 --- a/README_ja.md +++ b/README_ja.md @@ -132,7 +132,7 @@ pip install dingo-python # HHEM 幻覚検出モデル(transformers + torch が必要) pip install "dingo-python[hhem]" -# 検索プラットフォームベンチマーク評価(MTEB + pytrec_eval) +# 検索プラットフォームベンチマーク評価(MTEB + pytrec-eval-terrier) pip install "dingo-python[retrieval]" # 全機能をインストール(HHEM + Agent + Retrieval) diff --git a/README_zh-CN.md b/README_zh-CN.md index 7113b3de..a488df57 100644 --- a/README_zh-CN.md +++ b/README_zh-CN.md @@ -133,7 +133,7 @@ pip install dingo-python # 安装 HHEM 幻觉检测模型(需要 transformers + torch) pip install "dingo-python[hhem]" -# 安装检索平台评测(MTEB + pytrec_eval) +# 安装检索平台评测(MTEB + pytrec-eval-terrier) pip install "dingo-python[retrieval]" # 安装全部功能(HHEM + Agent + Retrieval) diff --git a/requirements/retrieval.txt b/requirements/retrieval.txt index b3ca2ff0..7ada5d08 100644 --- a/requirements/retrieval.txt +++ b/requirements/retrieval.txt @@ -1,2 +1,2 @@ mteb>=1.0.0 -pytrec_eval +pytrec-eval-terrier>=0.5.6