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

benchmark

Browse Source
This commit is contained in:
shitao 2023-08-05 20:06:27 +08:00
parent 9ab6b70e5e
commit 9e97e1895c
3 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmark/eval_MTEB.py
View File

@ -12,7 +12,7 @@ query_instruction_for_retrieval_dict = {
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument('--model_name_or_path', default="BAAI/bge-large-en", type=str)
parser.add_argument('--task_type', default=None, type=str)
parser.add_argument('--task_type', default=None, type=str, help="task type. Default is None, which means using all task types")
return parser.parse_args()

2
benchmark/setup.py
View File

@ -3,6 +3,8 @@ from setuptools import setup, find_packages
setup(
name='C_MTEB',
version='1.0.0',
author_email='2906698981@qq.com',
url='https://github.com/FlagOpen/FlagEmbedding/tree/master/benchmark',
package_dir={"": "C_MTEB"},
packages=find_packages("C_MTEB"),
install_requires=[

2
setup.py
View File

@ -3,6 +3,8 @@ from setuptools import setup, find_packages
setup(
name='flag_embedding',
version='1.0.0',
author_email='2906698981@qq.com',
url='https://github.com/FlagOpen/FlagEmbedding',
package_dir={"": "flag_embedding"},
packages=find_packages("flag_embedding"),
install_requires=[