mteb 2.4.1__py3-none-any.whl → 2.5.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (40) hide show
  1. mteb/benchmarks/benchmark.py +31 -13
  2. mteb/benchmarks/benchmarks/benchmarks.py +2 -2
  3. mteb/cache.py +36 -7
  4. mteb/descriptive_stats/Classification/TurkishConstitutionalCourtViolation.json +54 -0
  5. mteb/descriptive_stats/Retrieval/SQuADKorV1Retrieval.json +30 -0
  6. mteb/models/model_implementations/andersborges.py +12 -0
  7. mteb/models/model_implementations/bge_models.py +43 -0
  8. mteb/models/model_implementations/codefuse_models.py +144 -0
  9. mteb/models/model_implementations/dino_models.py +152 -0
  10. mteb/models/model_implementations/emillykkejensen_models.py +18 -0
  11. mteb/models/model_implementations/euler_models.py +6 -0
  12. mteb/models/model_implementations/fa_models.py +50 -0
  13. mteb/models/model_implementations/facebookai.py +44 -0
  14. mteb/models/model_implementations/gte_models.py +69 -0
  15. mteb/models/model_implementations/kalm_models.py +38 -0
  16. mteb/models/model_implementations/kblab.py +6 -0
  17. mteb/models/model_implementations/kowshik24_models.py +9 -0
  18. mteb/models/model_implementations/misc_models.py +293 -0
  19. mteb/models/model_implementations/mod_models.py +10 -23
  20. mteb/models/model_implementations/mxbai_models.py +6 -0
  21. mteb/models/model_implementations/nomic_models.py +8 -0
  22. mteb/models/model_implementations/nvidia_llama_nemoretriever_colemb.py +5 -3
  23. mteb/models/model_implementations/pylate_models.py +33 -0
  24. mteb/models/model_implementations/ru_sentence_models.py +22 -0
  25. mteb/models/model_implementations/seed_1_6_embedding_models_1215.py +658 -0
  26. mteb/models/model_implementations/sentence_transformers_models.py +39 -0
  27. mteb/models/model_implementations/spartan8806_atles_champion.py +7 -0
  28. mteb/models/model_implementations/ua_sentence_models.py +9 -0
  29. mteb/models/model_implementations/vi_vn_models.py +33 -0
  30. mteb/results/benchmark_results.py +22 -4
  31. mteb/tasks/classification/tur/__init__.py +4 -0
  32. mteb/tasks/classification/tur/turkish_constitutional_court.py +41 -0
  33. mteb/tasks/retrieval/kor/__init__.py +2 -1
  34. mteb/tasks/retrieval/kor/squad_kor_v1_retrieval.py +47 -0
  35. {mteb-2.4.1.dist-info → mteb-2.5.0.dist-info}/METADATA +1 -1
  36. {mteb-2.4.1.dist-info → mteb-2.5.0.dist-info}/RECORD +40 -35
  37. {mteb-2.4.1.dist-info → mteb-2.5.0.dist-info}/WHEEL +0 -0
  38. {mteb-2.4.1.dist-info → mteb-2.5.0.dist-info}/entry_points.txt +0 -0
  39. {mteb-2.4.1.dist-info → mteb-2.5.0.dist-info}/licenses/LICENSE +0 -0
  40. {mteb-2.4.1.dist-info → mteb-2.5.0.dist-info}/top_level.txt +0 -0
mteb/models/model_implementations/ru_sentence_models.py CHANGED
@@ -439,6 +439,13 @@ user_bge_m3 = ModelMeta(
439
439
  },
440
440
  public_training_code=None,
441
441
  public_training_data=None,
442
+ citation="""@misc{deepvk2024user,
443
+ title={USER: Universal Sentence Encoder for Russian},
444
+ author={Malashenko, Boris and Zemerov, Anton and Spirin, Egor},
445
+ url={https://huggingface.co/datasets/deepvk/USER-base},
446
+ publisher={Hugging Face},
447
+ year={2024},
448
+ }""",
442
449
  )
443
450
 
444
451
  deberta_v1_ru = ModelMeta(
@@ -873,6 +880,7 @@ frida = ModelMeta(
873
880
  public_training_data=None,
874
881
  public_training_code=None,
875
882
  framework=["Sentence Transformers", "PyTorch"],
883
+ citation=None,
876
884
  )
877
885
 
878
886
  giga_embeddings = ModelMeta(
@@ -1008,6 +1016,13 @@ user2_small = ModelMeta(
1008
1016
  public_training_data=None,
1009
1017
  public_training_code="https://github.com/BlessedTatonka/some_code/tree/2899f27d51efdf4217fc6453799ff197e9792f1e",
1010
1018
  framework=["Sentence Transformers", "PyTorch"],
1019
+ citation="""@misc{deepvk2025user,
1020
+ title={USER2},
1021
+ author={Malashenko, Boris and Spirin, Egor and Sokolov Andrey},
1022
+ url={https://huggingface.co/deepvk/USER2-small},
1023
+ publisher={Hugging Face},
1024
+ year={2025},
1025
+ }""",
1011
1026
  )
1012
1027
 
1013
1028
  user2_base = ModelMeta(
@@ -1033,4 +1048,11 @@ user2_base = ModelMeta(
1033
1048
  public_training_data=None,
1034
1049
  public_training_code="https://github.com/BlessedTatonka/some_code/tree/2899f27d51efdf4217fc6453799ff197e9792f1e",
1035
1050
  framework=["Sentence Transformers", "PyTorch"],
1051
+ citation="""@misc{deepvk2025user,
1052
+ title={USER2},
1053
+ author={Malashenko, Boris and Spirin, Egor and Sokolov Andrey},
1054
+ url={https://huggingface.co/deepvk/USER2-base},
1055
+ publisher={Hugging Face},
1056
+ year={2025},
1057
+ }""",
1036
1058
  )