Update training datasets of GeoGPT-Research-Project/GeoEmbedding model

mteb/models/geogpt_models.py

@@ -0,0 +1,68 @@
+"""Models for GeoGPT-Research-Project"""
+
+from __future__ import annotations
+
+from functools import partial
+
+import torch
+
+from mteb.model_meta import ModelMeta
+from mteb.models.instruct_wrapper import InstructSentenceTransformerWrapper
+
+geoembedding = ModelMeta(
+    name="GeoGPT-Research-Project/GeoEmbedding",
+    languages=["eng-Latn"],  # follows ISO 639-3 and BCP-47
+    open_weights=True,
+    revision="29803c28ea7ef6871194a8ebc85ad7bfe174928e",
+    loader=partial(
+        InstructSentenceTransformerWrapper,
+        "GeoGPT-Research-Project/GeoEmbedding",
+        "29803c28ea7ef6871194a8ebc85ad7bfe174928e",
+        instruction_template="Instruct: {instruction}\nQuery: ",
+        apply_instruction_to_passages=False,
+        model_kwargs={"torch_dtype": torch.bfloat16},
+        trust_remote_code=True,
+    ),
+    release_date="2025-04-22",
+    n_parameters=7241732096,
+    memory_usage_mb=27625,
+    embed_dim=4096,
+    license="apache-2.0",
+    max_tokens=32768,
+    reference="https://huggingface.co/GeoGPT-Research-Project/GeoEmbedding",
+    similarity_fn_name="cosine",
+    framework=["Sentence Transformers", "PyTorch"],
+    use_instructions=True,
+    public_training_code="",
+    public_training_data="",
+    training_datasets={
+        "ArguAna": ["test"],
+        "FEVER": ["train"],
+        "MSMARCO": ["train"],
+        "FiQA2018": ["train"],
+        "HotpotQA": ["train"],
+        "NFCorpus": ["train"],
+        "SciFact": ["train"],
+        "AmazonCounterfactualClassification": ["train"],
+        "AmazonPolarityClassification": ["train"],
+        "AmazonReviewsClassification": ["train"],
+        "Banking77Classification": ["train"],
+        "EmotionClassification": ["train"],
+        "MassiveIntentClassification": ["train"],
+        "MTOPDomainClassification": ["train"],
+        "MTOPIntentClassification": ["train"],
+        "ToxicConversationsClassification": ["train"],
+        "TweetSentimentExtractionClassification": ["train"],
+        "ArxivClusteringS2S": ["test"],
+        "ArxivClusteringP2P": ["test"],
+        "MedrixvClusteringS2S": ["test"],
+        "MedrixvClusteringP2P": ["test"],
+        "BiorxivClusteringS2S": ["test"],
+        "BiorxivClusteringP2P": ["test"],
+        "TwentyNewsgroupsClustering": ["test"], 
+        "STS12": ["train"],
+        "STS22": ["train"],
+        "STSBenchmark": ["train"],
+        "StackOverflowDupQuestions": ["train"],
+    },
+)

mteb/models/overview.py

@@ -88,6 +88,7 @@
     vlm2vec_models,
     voyage_models,
     voyage_v,
+    geogpt_models,
 )
 
 logger = logging.getLogger(__name__)
@@ -169,6 +170,7 @@
     colpali_models,
     colqwen_models,
     colsmol_models,
+    geogpt_models,
 ]
 MODEL_REGISTRY = {}