joe32140's picture
Add new SentenceTransformer model
3a4f119 verified
|
raw
history blame
30.4 kB
metadata
base_model: answerdotai/ModernBERT-base
datasets:
  - sentence-transformers/msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1
language:
  - en
library_name: sentence-transformers
metrics:
  - cosine_accuracy
pipeline_tag: sentence-similarity
tags:
  - sentence-transformers
  - sentence-similarity
  - feature-extraction
  - generated_from_trainer
  - dataset_size:11662655
  - loss:CachedMultipleNegativesRankingLoss
widget:
  - source_sentence: what county is lyndhurst, ohio in
    sentences:
      - >-
        This article is about the song written by Kenneth Gamble, Leon Huff and
        Cary Gilbert. For the Tina Turner song, see Don't Leave Me This Way
        (Tina Turner song). Don't Leave Me This Way is a song written by Kenneth
        Gamble, Leon Huff and Cary Gilbert. First charting as a hit for Harold
        Melvin & the Blue Notes featuring Teddy Pendergrass, an act on Gamble &
        Huff's Philadelphia International label in 1975, Don't Leave Me This Way
        was later a huge disco hit for Motown artist Thelma Houston in 1977.
      - "Lyndhurst is a city in Cuyahoga County, Ohio, United States. The population was 14,001 at the 2010 census. Lyndhurst is located in northeastern Ohio, and is a suburb of Cleveland. A small part of Lyndhurst was originally part of Mayfield Township. It used to be called Euclidville before Lyndhurst was chosen. Lyndhurst is located at 41°31â\x80²17â\x80³N 81°29â\x80²25â\x80³W / 41.52139°N 81.49028°W / 41.52139; -81.49028 (41.521352, -81.490141)."
      - >-
        Welcome to Trumbull County... Trumbull County, the county seat, located
        in Warren, Ohio, consists of a combination of both urban and rural
        communities situated in the northeast corner of Ohio. It is situated
        roughly between the Youngstown, Cleveland and Akron corridors.
  - source_sentence: who founded the american graphophone company
    sentences:
      - >-
        In 1886, Graham Bell and Charles Sumner Tainter founded the American
        Graphophone Company to distribute and sell graphophones in the US and
        Canada under license from the Volta Graphophone Company. In 1890, the
        American Graphophone Company stopped production of new phonographs due
        to sagging orders.
      - >-
        ShelfGenie How much does a ShelfGenie franchise cost? ShelfGenie has a
        franchise fee of up to $45,000, with a total initial investment range of
        $70,100 to $107,750. Local ShelfGenie franchise opportunities.
        ShelfGenie is looking to grow in a number of cities around the country.
        To find out if there's a franchise opportunity in your city, unlock more
        information.
      - "A+E Networks. The technology that made the modern music business possible came into existence in the New Jersey laboratory where Thomas Alva Edison created the first device to both record sound and play it back. He was awarded U.S. Patent No. 200,521 for his inventionâ\x80\x93the phonographâ\x80\x93on this day in 1878."
  - source_sentence: is housekeeping camp flooded?
    sentences:
      - >-
        What is the importance of housekeeping at work? A: Workplace
        housekeeping promotes sanitation, safety, organization and productivity.
        It also boosts morale. Daily housekeeping maintenance keeps the
        workplac... Full Answer >
      - >-
        The back patio area of a cabin is partially submerged in flood water at
        Housekeeping Camp on Monday, Jan. 9, 2017, in Yosemite National Park.
        The Merced River, swollen with storm runoff, crested at 12.7 feet at 4
        a.m. SILVIA FLORES sflores@fresnobee.com.
      - "1 Bake for 8 minutes, then rotate the pan and check the underside of the bagels. 2  If theyâ\x80\x99re getting too dark, place another pan under the baking sheet. ( 3 Doubling the pan will insulate the first baking sheet.) Bake for another 8 to 12 minutes, until the bagels are a golden brown. 4  13."
  - source_sentence: causes for infection in the nerve of tooth
    sentences:
      - >-
        If a cavity is causing the toothache, your dentist will fill the cavity
        or possibly extract the tooth, if necessary. A root canal might be
        needed if the cause of the toothache is determined to be an infection of
        the tooth's nerve. Bacteria that have worked their way into the inner
        aspects of the tooth cause such an infection. An antibiotic may be
        prescribed if there is fever or swelling of the jaw.
      - "According to Article III, Section 1 of the Constitution, judges and justices of the Judicial Branch serve during good behavior.. This means they are appointed for life, unles â\x80¦ s they are impeached and removed from office. + 50 others found this useful.he term length for members of the House are two years and a staggering six years for members of the Senate."
      - >-
        Inflamed or infected pulp (pulpitis) most often causes a toothache. To
        relieve the pain and prevent further complications, the tooth may be
        extracted (surgically removed) or saved by root canal treatment.
  - source_sentence: what county is hayden in
    sentences:
      - >-
        Normally, the Lead Agency is the agency with general governmental powers
        such as a city or a county. Agencies with limited powers or districts
        that provide a public service/utility such as a recreation and park
        district will tend to be a Responsible Agency.
      - >-
        According to the United States Census Bureau, the city has a total area
        of 9.61 square miles (24.89 km2), of which 9.60 square miles (24.86 km2)
        is land and 0.01 square miles (0.03 km2) is water. It lies at the
        southwestern end of Hayden Lake, and the elevation of the city is 2,287
        feet (697 m) above sea level. Hayden is located on U.S. Route 95 at the
        junction of Route 41. It is also four miles (6 km) north of Interstate
        90 and Coeur d'Alene. The Coeur d'Alene airport is northwest of Hayden.
      - >-
        Hayden is a city in Kootenai County, Idaho, United States. Located in
        the northern portion of the state, just north of Coeur d'Alene, its
        population was 13,294 at the 2010 census.
model-index:
  - name: SentenceTransformer based on answerdotai/ModernBERT-base
    results:
      - task:
          type: triplet
          name: Triplet
        dataset:
          name: msmarco co condenser dev
          type: msmarco-co-condenser-dev
        metrics:
          - type: cosine_accuracy
            value: 0.984
            name: Cosine Accuracy

SentenceTransformer based on answerdotai/ModernBERT-base

This is a sentence-transformers model finetuned from answerdotai/ModernBERT-base on the msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1 dataset. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.

Model Details

Model Description

Model Sources

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 8192, 'do_lower_case': False}) with Transformer model: ModernBertModel 
  (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
)

Usage

Direct Usage (Sentence Transformers)

First install the Sentence Transformers library:

pip install -U sentence-transformers

Then you can load this model and run inference.

from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("joe32140/ModernBERT-base-DPR-8e-05")
# Run inference
sentences = [
    'what county is hayden in',
    "Hayden is a city in Kootenai County, Idaho, United States. Located in the northern portion of the state, just north of Coeur d'Alene, its population was 13,294 at the 2010 census.",
    "According to the United States Census Bureau, the city has a total area of 9.61 square miles (24.89 km2), of which 9.60 square miles (24.86 km2) is land and 0.01 square miles (0.03 km2) is water. It lies at the southwestern end of Hayden Lake, and the elevation of the city is 2,287 feet (697 m) above sea level. Hayden is located on U.S. Route 95 at the junction of Route 41. It is also four miles (6 km) north of Interstate 90 and Coeur d'Alene. The Coeur d'Alene airport is northwest of Hayden.",
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]

Evaluation

Metrics

Triplet

Metric Value
cosine_accuracy 0.984

Training Details

Training Dataset

msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1

  • Dataset: msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1 at 84ed2d3
  • Size: 11,662,655 training samples
  • Columns: query, positive, and negative
  • Approximate statistics based on the first 1000 samples:
    query positive negative
    type string string string
    details
    • min: 4 tokens
    • mean: 9.26 tokens
    • max: 34 tokens
    • min: 17 tokens
    • mean: 79.14 tokens
    • max: 222 tokens
    • min: 24 tokens
    • mean: 80.09 tokens
    • max: 436 tokens
  • Samples:
    query positive negative
    what is the meaning of menu planning Menu planning is the selection of a menu for an event. Such as picking out the dinner for your wedding or even a meal at a Birthday Party. Menu planning is when you are preparing a calendar of meals and you have to sit down and decide what meat and veggies you want to serve on each certain day. Menu Costs. In economics, a menu cost is the cost to a firm resulting from changing its prices. The name stems from the cost of restaurants literally printing new menus, but economists use it to refer to the costs of changing nominal prices in general.
    how old is brett butler Brett Butler is 59 years old. To be more precise (and nerdy), the current age as of right now is 21564 days or (even more geeky) 517536 hours. That's a lot of hours! Passed in: St. John's, Newfoundland and Labrador, Canada. Passed on: 16/07/2016. Published in the St. John's Telegram. Passed away suddenly at the Health Sciences Centre surrounded by his loving family, on July 16, 2016 Robert (Bobby) Joseph Butler, age 52 years. Predeceased by his special aunt Geri Murrin and uncle Mike Mchugh; grandparents Joe and Margaret Murrin and Jack and Theresa Butler.
    when was the last navajo treaty sign? In Executive Session, Senate of the United States, July 25, 1868. Resolved, (two-thirds of the senators present concurring,) That the Senate advise and consent to the ratification of the treaty between the United States and the Navajo Indians, concluded at Fort Sumner, New Mexico, on the first day of June, 1868. Share Treaty of Greenville. The Treaty of Greenville was signed August 3, 1795, between the United States, represented by Gen. Anthony Wayne, and chiefs of the Indian tribes located in the Northwest Territory, including the Wyandots, Delawares, Shawnees, Ottawas, Miamis, and others.
  • Loss: CachedMultipleNegativesRankingLoss with these parameters:
    {
        "scale": 20.0,
        "similarity_fct": "cos_sim"
    }
    

Evaluation Dataset

msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1

  • Dataset: msmarco-co-condenser-margin-mse-sym-mnrl-mean-v1 at 84ed2d3
  • Size: 11,662,655 evaluation samples
  • Columns: query, positive, and negative
  • Approximate statistics based on the first 1000 samples:
    query positive negative
    type string string string
    details
    • min: 4 tokens
    • mean: 9.2 tokens
    • max: 27 tokens
    • min: 21 tokens
    • mean: 80.44 tokens
    • max: 241 tokens
    • min: 23 tokens
    • mean: 80.38 tokens
    • max: 239 tokens
  • Samples:
    query positive negative
    what county is holly springs nc in Holly Springs, North Carolina. Holly Springs is a town in Wake County, North Carolina, United States. As of the 2010 census, the town population was 24,661, over 2½ times its population in 2000. Contents. The Mt. Holly Springs Park & Resort. One of the numerous trolley routes that carried people around the county at the turn of the century was the Carlisle & Mt. Holly Railway Company. The “Holly Trolley” as it came to be known was put into service by Patricio Russo and made its first run on May 14, 1901.
    how long does nyquil stay in your system In order to understand exactly how long Nyquil lasts, it is absolutely vital to learn about the various ingredients in the drug. One of the ingredients found in Nyquil is Doxylamine, which is an antihistamine. This specific medication has a biological half-life or 6 to 12 hours. With this in mind, it is possible for the drug to remain in the system for a period of 12 to 24 hours. It should be known that the specifics will depend on a wide variety of different factors, including your age and metabolism. I confirmed that NyQuil is about 10% alcohol, a higher content than most domestic beers. When I asked about the relatively high proof, I was told that the alcohol dilutes the active ingredients. The alcohol free version is there for customers with addiction issues.. also found that in that version there is twice the amount of DXM. When I asked if I could speak to a chemist or scientist, I was told they didn't have anyone who fit that description there. It’s been eight years since I kicked NyQuil. I've been sober from alcohol for four years.
    what are mineral water 1 Mineral water – water from a mineral spring that contains various minerals, such as salts and sulfur compounds. 2 It comes from a source tapped at one or more bore holes or spring, and originates from a geologically and physically protected underground water source. Mineral water – water from a mineral spring that contains various minerals, such as salts and sulfur compounds. 2 It comes from a source tapped at one or more bore holes or spring, and originates from a geologically and physically protected underground water source. Minerals for Your Body. Drinking mineral water is beneficial to health and well-being. But it is not only the amount of water you drink that is important-what the water contains is even more essential.inerals for Your Body. Drinking mineral water is beneficial to health and well-being. But it is not only the amount of water you drink that is important-what the water contains is even more essential.
  • Loss: CachedMultipleNegativesRankingLoss with these parameters:
    {
        "scale": 20.0,
        "similarity_fct": "cos_sim"
    }
    

Training Hyperparameters

Non-Default Hyperparameters

  • per_device_train_batch_size: 512
  • per_device_eval_batch_size: 512
  • learning_rate: 8e-05
  • num_train_epochs: 1
  • warmup_ratio: 0.05
  • bf16: True
  • batch_sampler: no_duplicates

All Hyperparameters

Click to expand
  • overwrite_output_dir: False
  • do_predict: False
  • eval_strategy: no
  • prediction_loss_only: True
  • per_device_train_batch_size: 512
  • per_device_eval_batch_size: 512
  • per_gpu_train_batch_size: None
  • per_gpu_eval_batch_size: None
  • gradient_accumulation_steps: 1
  • eval_accumulation_steps: None
  • torch_empty_cache_steps: None
  • learning_rate: 8e-05
  • weight_decay: 0.0
  • adam_beta1: 0.9
  • adam_beta2: 0.999
  • adam_epsilon: 1e-08
  • max_grad_norm: 1.0
  • num_train_epochs: 1
  • max_steps: -1
  • lr_scheduler_type: linear
  • lr_scheduler_kwargs: {}
  • warmup_ratio: 0.05
  • warmup_steps: 0
  • log_level: passive
  • log_level_replica: warning
  • log_on_each_node: True
  • logging_nan_inf_filter: True
  • save_safetensors: True
  • save_on_each_node: False
  • save_only_model: False
  • restore_callback_states_from_checkpoint: False
  • no_cuda: False
  • use_cpu: False
  • use_mps_device: False
  • seed: 42
  • data_seed: None
  • jit_mode_eval: False
  • use_ipex: False
  • bf16: True
  • fp16: False
  • fp16_opt_level: O1
  • half_precision_backend: auto
  • bf16_full_eval: False
  • fp16_full_eval: False
  • tf32: None
  • local_rank: 0
  • ddp_backend: None
  • tpu_num_cores: None
  • tpu_metrics_debug: False
  • debug: []
  • dataloader_drop_last: False
  • dataloader_num_workers: 0
  • dataloader_prefetch_factor: None
  • past_index: -1
  • disable_tqdm: False
  • remove_unused_columns: True
  • label_names: None
  • load_best_model_at_end: False
  • ignore_data_skip: False
  • fsdp: []
  • fsdp_min_num_params: 0
  • fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
  • fsdp_transformer_layer_cls_to_wrap: None
  • accelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
  • deepspeed: None
  • label_smoothing_factor: 0.0
  • optim: adamw_torch
  • optim_args: None
  • adafactor: False
  • group_by_length: False
  • length_column_name: length
  • ddp_find_unused_parameters: None
  • ddp_bucket_cap_mb: None
  • ddp_broadcast_buffers: False
  • dataloader_pin_memory: True
  • dataloader_persistent_workers: False
  • skip_memory_metrics: True
  • use_legacy_prediction_loop: False
  • push_to_hub: False
  • resume_from_checkpoint: None
  • hub_model_id: None
  • hub_strategy: every_save
  • hub_private_repo: None
  • hub_always_push: False
  • gradient_checkpointing: False
  • gradient_checkpointing_kwargs: None
  • include_inputs_for_metrics: False
  • include_for_metrics: []
  • eval_do_concat_batches: True
  • fp16_backend: auto
  • push_to_hub_model_id: None
  • push_to_hub_organization: None
  • mp_parameters:
  • auto_find_batch_size: False
  • full_determinism: False
  • torchdynamo: None
  • ray_scope: last
  • ddp_timeout: 1800
  • torch_compile: False
  • torch_compile_backend: None
  • torch_compile_mode: None
  • dispatch_batches: None
  • split_batches: None
  • include_tokens_per_second: False
  • include_num_input_tokens_seen: False
  • neftune_noise_alpha: None
  • optim_target_modules: None
  • batch_eval_metrics: False
  • eval_on_start: False
  • use_liger_kernel: False
  • eval_use_gather_object: False
  • average_tokens_across_devices: False
  • prompts: None
  • batch_sampler: no_duplicates
  • multi_dataset_batch_sampler: proportional

Training Logs

Epoch Step Training Loss msmarco-co-condenser-dev_cosine_accuracy
0 0 - 0.606
0.2048 500 0.6405 -
0.4095 1000 0.1473 -
0.6143 1500 0.1091 -
0.8190 2000 0.0907 -
1.0 2442 - 0.984

Framework Versions

  • Python: 3.11.9
  • Sentence Transformers: 3.3.1
  • Transformers: 4.48.0.dev0
  • PyTorch: 2.4.0
  • Accelerate: 1.2.1
  • Datasets: 2.21.0
  • Tokenizers: 0.21.0

Citation

BibTeX

Sentence Transformers

@inproceedings{reimers-2019-sentence-bert,
    title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
    author = "Reimers, Nils and Gurevych, Iryna",
    booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
    month = "11",
    year = "2019",
    publisher = "Association for Computational Linguistics",
    url = "https://arxiv.org/abs/1908.10084",
}

CachedMultipleNegativesRankingLoss

@misc{gao2021scaling,
    title={Scaling Deep Contrastive Learning Batch Size under Memory Limited Setup},
    author={Luyu Gao and Yunyi Zhang and Jiawei Han and Jamie Callan},
    year={2021},
    eprint={2101.06983},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}