Task
Feature Extraction
B
BAAI/bge-small-en
Feature Extraction·BAAI· 677.7K· 93
transformers
mit
Feature Extraction
33.4M params
arxiv:2311.13534arxiv:2310.07554arxiv:2309.07597license:mit
Recommend switching to newest BAAI/bge-small-en-v1.5, which has more reasonable similarity distribution and same method of usage.
# pull & run locally
pip install mlforge-sdk && mlforge pull BAAI/bge-small-en
pip install mlforge-sdk && mlforge pull BAAI/bge-small-en
Model details
Provider
BAAI
Framework
transformers
Parameters
33.4M
Size
381 MB
License
mit
Downloads
677.7K
Likes
93
Paper
arXiv:2311.13534
Updated
2023-12-13
About BAAI/bge-small-en
--- tags: - mteb - sentence transformers model-index: - name: bge-small-en results: - task: type: Classification dataset: type: mteb/amazoncounterfactual name: MTEB AmazonCounterfactualClassification (en) config: en split: test revision: e8379541af4e31359cca9fbcf4b00f2671dba205 metrics: - type: accuracy value: 74.34328358208955 - type: ap value: 37.59947775195661 - type: f1 value: 68.548415491933 - task: type: Classification dataset: type: mteb/amazonpolarity name: MTEB AmazonPolarityClassification config: default split: test revision: e2d317d38cd51312af73b3d32a06d1a08b442046 metrics: - type: accuracy value: 93.04527499999999 - type: ap value: 89.60696356772135 - type: f1 value: 93.03361469382438 - task: type: Classification dataset: type: mteb/amazonreviewsmulti name: MTEB AmazonReviewsClassification (en) config: en split: test revision: 1399c76144fd37290681b995c656ef9b2e06e26d metrics: - type: accuracy value: 46.08 - type: f1 value: 45.66249835363254 - task: type: Retrieval dataset: type: arguana name: MTEB ArguAna config: default split: test revision: None metrics: - type: mapat1 value: 35.205999999999996 - type: mapat10 value: 50.782000000000004 - type: mapat100 value: 51.547 - type: mapat1000 value: 51.554 - type: mapat3 value: 46.515 - type: mapat5