yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2025-06-27 02:39:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1359 from hanhainebula/master

Browse Source 
Fix a bug in beir evaluation and release v1.3.4
This commit is contained in:
Jianlyu Chen 2025-02-07 20:28:28 +08:00 committed by GitHub
commit 0306f566ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
FlagEmbedding/evaluation/beir/evaluator.py
View File

@ -87,7 +87,8 @@ class BEIREvaluator(AbsEvaluator):
if sub_dataset_names is None:
if dataset_name is not None:
save_name = f"{dataset_name}-" + "{split}.json"
corpus_embd_save_dir = os.path.join(corpus_embd_save_dir, str(retriever), dataset_name)
if corpus_embd_save_dir is not None:
corpus_embd_save_dir = os.path.join(corpus_embd_save_dir, str(retriever), dataset_name)
else:
save_name = "{split}.json"
@ -213,7 +214,8 @@ class BEIREvaluator(AbsEvaluator):
for sub_dataset_name in sub_dataset_names:
if dataset_name is not None:
save_name = f"{dataset_name}-{sub_dataset_name}-" + "{split}.json"
corpus_embd_save_dir = os.path.join(corpus_embd_save_dir, str(retriever), dataset_name, sub_dataset_name)
if corpus_embd_save_dir is not None:
corpus_embd_save_dir = os.path.join(corpus_embd_save_dir, str(retriever), dataset_name, sub_dataset_name)
else:
save_name = f"{sub_dataset_name}-" + "{split}.json"

4
setup.py
View File

@ -5,7 +5,7 @@ with open("README.md", mode="r", encoding="utf-8") as readme_file:
setup(
name='FlagEmbedding',
version='1.3.3',
version='1.3.4',
description='FlagEmbedding',
long_description=readme,
long_description_content_type="text/markdown",
@ -16,7 +16,7 @@ setup(
install_requires=[
'torch>=1.6.0',
'transformers>=4.44.2',
'datasets==2.19.0',
'datasets>=2.19.0',
'accelerate>=0.20.1',
'sentence_transformers',
'peft',