all-MiniLM-L6-v48-pair_score

This is a sentence-transformers model finetuned from KhaledReda/all-MiniLM-L6-v36-pair_score on the pairs_with_scores_v40 dataset. It maps sentences & paragraphs to a 384-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 Type: Sentence Transformer
• Base model: KhaledReda/all-MiniLM-L6-v36-pair_score
• Maximum Sequence Length: 256 tokens
• Output Dimensionality: 384 dimensions
• Similarity Function: Cosine Similarity
• Training Dataset:
  • pairs_with_scores_v40
• Language: en
• License: apache-2.0

Model Sources

• Documentation: Sentence Transformers Documentation
• Repository: Sentence Transformers on GitHub
• Hugging Face: Sentence Transformers on Hugging Face

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 256, 'do_lower_case': False, 'architecture': 'BertModel'})
  (1): Pooling({'word_embedding_dimension': 384, '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})
  (2): Normalize()
)

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("sentence_transformers_model_id")
# Run inference
sentences = [
    'flipflops',
    'sashimi salmon - 5 pcs sashimi 5 pcs',
    'leggings flack seamless italian yarn silky seamless fabric shaperetaining flexible high elasticity flack seamless women gym strikers flack seamless s yarn fibers fabric knit green custom sourced italian yarn fibers are blended together to create an irresistibly silky smooth hand-feel. our exclusive full-interlock knit construction forms a more dense seamless fabric that prevents sheerness and maintains breathability. increased elasticity provides unmatched compression shape retention and flexibility made to move in sync with you. our physique enhancing confidence inducing seam work is structurally reinforced to provide support and durability.',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 384]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities)
# tensor([[ 1.0000, -0.0346,  0.1565],
#         [-0.0346,  1.0000, -0.0185],
#         [ 0.1565, -0.0185,  1.0000]])

Training Details

Training Dataset

pairs_with_scores_v40

• Dataset: pairs_with_scores_v40 at 41c251b
• Size: 19,507,210 training samples
• Columns: sentence1, sentence2, and score
• Approximate statistics based on the first 1000 samples:

  	sentence1	sentence2	score
  type	string	string	float
  details	min: 3 tokens mean: 6.53 tokens max: 43 tokens	min: 4 tokens mean: 61.73 tokens max: 256 tokens	min: 0.0 mean: 0.02 max: 1.0

• Samples:

  sentence1	sentence2	score
  arcos niza kitchen - 28.4cm	print making accessory patch n play fun tee patches accessory kids patches unisex patches boys patches girls patches velcro patches cloth patches our patch n play fun patches a delightful way to customize and personalize your child s clothing and accessories with our unique selection of patches you can pick the perfect designs that resonate with your child s personality and interests it come with velcro from the back to stick on the our patch n play fun tee	0.0
  sunny side up towel	pants blue straight swants summer melton breathable fabric pockets unisex loose fit men women unisex women men swants s drawstring pockets straight loose cotton melton blue summer egypt by iys - aka swants made for your comfort in super lazy days which is everyday use lol. made out of a cool cotton summer melton material and a wide bottom legs opening.unisex loose fit - check size chart for exact dimensions.- male model wearing size large- female model wearing size medium- designed amp produced in cairo egypt.	0.0
  arm blood pressure monitor	face powder maybelline fit me matte powd. mirror/110 fit me matte powder maybelline fit me matte powder shade maybelline fit me matte powder with mirror maybelline powder	0.0

• Loss: CoSENTLoss with these parameters:

  {
      "scale": 20.0,
      "similarity_fct": "pairwise_cos_sim"
  }
  

Evaluation Dataset

pairs_with_scores_v40

• Dataset: pairs_with_scores_v40 at 41c251b
• Size: 98,027 evaluation samples
• Columns: sentence1, sentence2, and score
• Approximate statistics based on the first 1000 samples:

  	sentence1	sentence2	score
  type	string	string	float
  details	min: 3 tokens mean: 6.36 tokens max: 25 tokens	min: 3 tokens mean: 61.43 tokens max: 256 tokens	min: 0.0 mean: 0.02 max: 1.0

• Samples:

  sentence1	sentence2	score
  heatresistant stove mitts	necklace original chain original women urban jewel chain design casual formal gold rock any outfit with the original chain this versatile piece is sure to make a statement with its classic design. perfect for any occasion you can t go wrong with this must-have accessory. add some flair to your wardrobe with this unique .	0.0
  glitter	four freezer bags freezer bag store your favorite foods with confidence using four freezer bags.	0.0
  cheese	risovi black quinoa seeds - 500 gr quinoa fiber quinoa seeds protein quinoa seeds folate quinoa seeds magnesium quinoa seeds gluten free quinoa seeds risovi quinoa seeds if you want a great and rich addition to your meals then quinoa from risovi will be a perfect choice. it s loaded with many important nutrients including fiber protein folate and magnesium. also it s gluten-free and suitable for people with gluten intolerance. add it to your breakfast cereals or yogurt. it can also be added to smoothies muffins pancakes and salads	0.0

• Loss: CoSENTLoss with these parameters:

  {
      "scale": 20.0,
      "similarity_fct": "pairwise_cos_sim"
  }
  

Training Hyperparameters

Non-Default Hyperparameters

• eval_strategy: steps
• per_device_train_batch_size: 64
• per_device_eval_batch_size: 64
• learning_rate: 2e-05
• num_train_epochs: 1
• warmup_ratio: 0.1
• fp16: True

All Hyperparameters

Click to expand

• overwrite_output_dir: False
• do_predict: False
• eval_strategy: steps
• prediction_loss_only: True
• per_device_train_batch_size: 64
• per_device_eval_batch_size: 64
• 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: 2e-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.1
• 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: False
• fp16: True
• 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
• hub_revision: None
• 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
• 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
• liger_kernel_config: None
• eval_use_gather_object: False
• average_tokens_across_devices: False
• prompts: None
• batch_sampler: batch_sampler
• multi_dataset_batch_sampler: proportional
• router_mapping: {}
• learning_rate_mapping: {}

Training Logs

Click to expand

Framework Versions

• Python: 3.12.3
• Sentence Transformers: 5.1.0
• Transformers: 4.55.4
• PyTorch: 2.6.0+cu124
• Accelerate: 1.10.1
• Datasets: 4.0.0
• Tokenizers: 0.21.4

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",
}

CoSENTLoss

@online{kexuefm-8847,
    title={CoSENT: A more efficient sentence vector scheme than Sentence-BERT},
    author={Su Jianlin},
    year={2022},
    month={Jan},
    url={https://kexue.fm/archives/8847},
}