README.md

metadata

tags:
  - sentence-transformers
  - sentence-similarity
  - feature-extraction
  - generated_from_trainer
  - dataset_size:114699
  - loss:CachedGISTEmbedLoss
base_model: BAAI/bge-large-en-v1.5
widget:
  - source_sentence: >-
      Bus drivers, including those operating in various sectors like public
      transit, intercity, private, or school services, need strong driving
      skills, knowledge of traffic laws, and the ability to operate safely in
      diverse conditions. Additionally, effective communication skills and the
      ability to handle passenger inquiries and emergencies are crucial.

      ['bus driver', 'intercity bus driver', 'private bus operator', 'transit
      bus driver', 'public service vehicle operator', 'passenger driver',
      'international bus driver', 'public bus operator', 'touristic bus driver',
      'coach driver', 'private coach driver', 'public bus driver', 'bus
      operator', 'driver of bus', 'bus driving operator', 'schoolbus driver']
    sentences:
      - >-
        The skill of determining shreds sizes percentage in cigarettes is
        primarily required by tobacco processing technicians and quality control
        specialists in the cigarette manufacturing industry, who ensure that the
        tobacco shreds meet specific size and quality standards for consistent
        product performance.

        ['determine shreds sizes percentage in cigarettes', 'determine shreds
        sizes percentage in cigarettes', 'determine the shreds sizes percentage
        of cigarettes', 'determine shreds size percentages in cigarettes',
        'agree shreds sizes percentage in cigarettes', 'determine the shreds
        sizes percentage in cigarettes', 'confirm shreds sizes percentage in
        cigarettes', 'sort shreds sizes percentage in cigarettes']
      - >-
        Job roles such as curriculum developers, educational consultants, and
        instructional designers require skills like analyzing, evaluating, and
        scrutinizing curriculums to improve educational outcomes. For
        legislative programmes, roles including policy analysts, legislative
        aides, and compliance officers use skills to test, evaluate, and
        scrutinize legislative processes to ensure effective and efficient
        policy implementation.

        ['analyse curriculum', 'test legislative programmes', 'evaluate
        legislative programmes', 'evaluate curriculum', 'test curriculum',
        'investigate curriculum', 'scrutinise curriculum', 'analyze 
        curriculum', 'scrutinise legislative processes', 'investigate
        legislative programmes']
      - >-
        Job roles such as customer service representatives, flight attendants,
        and hotel concierges require a strong focus on passengers or customers,
        ensuring their needs and comfort are prioritized to provide excellent
        service and support.

        ['focus on passengers', 'prioritise passengers', 'ensure passenger
        prioritisation', 'make passengers a priority', 'maintain a focus on
        passengers', 'ensure passengers are the priority focus', 'ensure
        passengers are prioritised', 'attend to passengers', 'ensure a focus on
        passengers']
  - source_sentence: >-
      A medical laboratory assistant, or any of its synonyms such as a
      biomedical laboratory assistant, requires strong attention to detail,
      proficiency in using laboratory equipment, and a foundational
      understanding of medical science. Additionally, skills in sample handling,
      data recording, and basic research methodologies are crucial for roles
      like a clinical research assistant or an assistant in medical laboratory.

      ['medical laboratory assistant', 'medical laboratory research assistant',
      'biomedical laboratory assistant', 'clinical research assistant',
      'assistant in medical laboratory', 'biomedical laboratory research
      assistant', 'assistant clinical researcher', 'medical lab assistant',
      'assistant in biomedical laboratory']
    sentences:
      - >-
        Job roles such as automotive mechanics, fleet managers, and vehicle
        technicians require skills to ensure vehicle operability and regular
        maintenance, which involves diagnosing and repairing issues to keep
        vehicles roadworthy and operational.

        ['ensure vehicle operability', 'keep vehicle roadworthy', 'keep vehicle
        operational', 'ensure operability of the vehicle', 'ensure vehicle
        remains operational', 'ensure maintenance of vehicle', 'ensure regular
        vehicle maintenance', 'ensure operation of the vehicle', 'ensure
        operability']
      - >-
        The skill of classroom management is primarily required by teachers and
        educators at all levels, from kindergarten to higher education, to
        ensure a productive, safe, and organized learning environment. It
        involves maintaining discipline, organizing space and materials, and
        facilitating effective instruction, roles that are crucial for teaching
        assistants and substitute teachers as well.

        ['perform classroom management', 'performing classroom management',
        'conduct classroom management', 'practice classroom management', 'carry
        out classroom management', 'implement classroom management', 'performs
        classroom management']
      - >-
        Job roles requiring expertise in stem cells, including embryonic and
        adult stem cells, typically include stem cell researchers, regenerative
        medicine scientists, and biomedical engineers who focus on the
        development and application of stem cell technologies for therapeutic
        purposes. Additionally, clinical researchers and medical practitioners
        in specialized fields such as oncology and hematology may utilize
        knowledge of stem cells for treatment and research purposes.

        ['stem cells', 'undifferentiated biological cells', 'embryonic stem
        cells', 'development of stem cells', 'stem cell', 'adult stem cells',
        'stem cells']
  - source_sentence: >-
      For roles such as 'physiotherapist', 'neuromusculoskeletal
      physiotherapist', 'osteopath', and 'chiropractor', the skills needed
      include a deep understanding of human anatomy and physiology, strong
      diagnostic skills, and the ability to apply manual therapy techniques to
      treat musculoskeletal issues. Additionally, effective communication skills
      are crucial for explaining treatments and exercises to patients, while
      adaptability and problem-solving skills are essential for tailoring
      treatments to individual patient needs.

      ['physiotherapist', 'neuromusculoskeletal physiotherapist', 'osteopath',
      'eurythmy therapist', 'respiratory therapist', 'remedial physiotherapist',
      'physiotherapist manager', 'occupational therapist', 'neurological
      physiotherapist', 'occupational physiotherapist', 'bobath
      physiotherapist', 'neuromuscular physiotherapist', 'manipulative
      physiotherapist', 'hydrotherapist', 'rehabilitation therapist',
      'masseuse', 'health promotion worker', 'cardiovascular physiotherapist',
      'respiratory physiotherapist', 'chiropractor', 'sports physiotherapist',
      'chiropractic therapist', 'neurodevelopmental physiotherapist', 'physical
      therapist', 'health and well-being therapist', 'business physiotherapist']
    sentences:
      - >-
        Job roles that require skills in dealing with emergency care situations
        include emergency medical technicians (EMTs), paramedics, and emergency
        room nurses or doctors, all of whom must quickly and effectively manage
        critical health situations to save lives.

        ['deal with emergency care situations', 'deal with emergency care
        situation', 'handle emergency care situation', 'apply knowledge in
        emergency care situations', 'handle emergency care situations']
      - >-
        Job roles such as fashion designers, stylist coordinators, and jewelry
        designers require the skill to distinguish and evaluate accessories,
        their differences, and applications, to ensure the right aesthetic and
        functional fit for their designs or clients. This skill is crucial for
        creating cohesive looks and enhancing the overall visual appeal in
        fashion and design industries.

        ['distinguish accessories', 'evaluate accessories and their
        differences', 'evaluate accessories and their application',
        'differentiate accessories', 'distinguish accessories and their
        application', 'distinguish differences in accessories']
      - >-
        Job roles that require expertise in curriculum objectives include
        educational consultants, curriculum developers, and instructional
        designers, who are tasked with creating and refining educational content
        and learning goals to meet specific educational standards and student
        needs. Teachers and headteachers also utilize these skills to align
        their teaching methods and materials with the set educational targets
        and aims.

        ['curriculum objectives', 'curriculum objective', 'curriculum goals',
        'curriculum targets', 'curriculum aims', 'curricula objectives']
  - source_sentence: >-
      A mine surveyor, also known as a mining surveyor or mine planning
      surveyor, requires expertise in geomatics and mining engineering to
      accurately map and plan mine operations, ensuring safety and efficiency.
      They must also possess strong analytical skills and the ability to use
      specialized software for creating detailed mine plans and maintaining
      accurate records.

      ['mine surveyor', 'mining surveyor', 'mine operations surveyor', 'mine
      plan maker', 'mine records keeper', 'mine surveyors', 'planner of mining
      operations', 'mine planning surveyor']
    sentences:
      - >-
        Job roles such as data analysts, business analysts, and financial
        analysts require the skill to present reports or prepare statistical
        reports, as they often need to communicate complex data insights clearly
        and effectively to stakeholders.

        ['present reports', 'present a report', 'submit presentation', 'prepare
        statistical reports']
      - >-
        Job roles such as Food Safety Manager, Quality Assurance Specialist, and
        Public Health Inspector require the skill of developing food safety
        programs to ensure compliance with regulations and maintain high
        standards of food safety in various settings including manufacturing,
        retail, and public health sectors.

        ['develop food safety programmes', 'creating food safety programmes',
        'develop programmes for food safety', 'food safety programmes creating',
        'food safety programmes developing', 'develop food safety programs',
        'food safety programme developing', 'food safety programme creating',
        'create food safety programmes', 'create programmes for food safety',
        'developing food safety programmes']
      - >-
        The skill of using a sander, whether it be a handheld, manual,
        automatic, or drywall sander, is primarily required by construction
        workers, carpenters, and drywall installers for tasks such as roughening
        and smoothing wall surfaces to prepare them for painting or finishing.

        ['use sander', 'use handheld sander', 'roughening of wall surfaces',
        'use drywall sander', 'sanding of wall surfaces', 'using sander',
        'sander usage', 'use manual sander', 'drywall sanding', 'use automatic
        sander']
  - source_sentence: >-
      An insulation supervisor, regardless of the specific type of insulation
      material or installation area, requires strong project management skills,
      knowledge of building codes and safety regulations, and expertise in
      insulation techniques to oversee the installation process effectively and
      ensure quality standards are met.

      ['insulation supervisor', 'supervisor of installation of insulating
      materials', 'supervisor of insulation materials installation', 'supervisor
      of installation of insulation', 'solid wall insulation installation
      supervisor', 'insulation installers supervisor', 'cavity wall insulation
      installation supervisor', 'loft insulation installation supervisor']
    sentences:
      - >-
        Job roles such as Food Safety Inspector, Public Health Officer, and
        Environmental Health Specialist require the skill of taking action on
        food safety violations to ensure compliance with health regulations and
        maintain public safety standards.

        ['take action on food safety violations', 'invoke action on food safety
        violations', 'agree action on food safety violations', 'pursue action on
        food safety violations', 'determine action on food safety violations']
      - >-
        Job roles that require skills in operating and supervising textile
        printing machines include Textile Printer Operators, Printing Machine
        Technicians, and Textile Production Specialists. These roles involve
        setting up, running, and maintaining printing machinery to ensure
        high-quality textile printing.

        ['tend textile printing machines', 'activate and supervise printing
        machines for textile material', 'activate and supervise textile printing
        machines', 'tend printing machines for textile', 'tend printing machines
        for textile material', 'care for textile printing machines', 'operate
        printing machines for textile material', 'operate textile printing
        machines']
      - >-
        The skill of installing insulation material is primarily required by job
        roles such as insulation workers, HVAC technicians, and construction
        specialists, who are responsible for improving energy efficiency and
        thermal comfort in buildings by correctly fitting and fixing insulation
        materials in various structures.

        ['install insulation material', 'insulate structure', 'fix insulation',
        'insulation material installation', 'installation of insulation
        material', 'fitting insulation', 'insulating structure', 'installing
        insulation material', 'fixing insulation', 'fit insulation']
pipeline_tag: sentence-similarity
library_name: sentence-transformers
metrics:
  - cosine_accuracy@1
  - cosine_accuracy@20
  - cosine_accuracy@50
  - cosine_accuracy@100
  - cosine_accuracy@150
  - cosine_accuracy@200
  - cosine_precision@1
  - cosine_precision@20
  - cosine_precision@50
  - cosine_precision@100
  - cosine_precision@150
  - cosine_precision@200
  - cosine_recall@1
  - cosine_recall@20
  - cosine_recall@50
  - cosine_recall@100
  - cosine_recall@150
  - cosine_recall@200
  - cosine_ndcg@1
  - cosine_ndcg@20
  - cosine_ndcg@50
  - cosine_ndcg@100
  - cosine_ndcg@150
  - cosine_ndcg@200
  - cosine_mrr@1
  - cosine_mrr@20
  - cosine_mrr@50
  - cosine_mrr@100
  - cosine_mrr@150
  - cosine_mrr@200
  - cosine_map@1
  - cosine_map@20
  - cosine_map@50
  - cosine_map@100
  - cosine_map@150
  - cosine_map@200
  - cosine_map@500
model-index:
  - name: SentenceTransformer based on BAAI/bge-large-en-v1.5
    results:
      - task:
          type: information-retrieval
          name: Information Retrieval
        dataset:
          name: full en
          type: full_en
        metrics:
          - type: cosine_accuracy@1
            value: 0.7203947368421053
            name: Cosine Accuracy@1
          - type: cosine_accuracy@20
            value: 1
            name: Cosine Accuracy@20
          - type: cosine_accuracy@50
            value: 1
            name: Cosine Accuracy@50
          - type: cosine_accuracy@100
            value: 1
            name: Cosine Accuracy@100
          - type: cosine_accuracy@150
            value: 1
            name: Cosine Accuracy@150
          - type: cosine_accuracy@200
            value: 1
            name: Cosine Accuracy@200
          - type: cosine_precision@1
            value: 0.7203947368421053
            name: Cosine Precision@1
          - type: cosine_precision@20
            value: 0.4916118421052631
            name: Cosine Precision@20
          - type: cosine_precision@50
            value: 0.3869736842105263
            name: Cosine Precision@50
          - type: cosine_precision@100
            value: 0.3046052631578947
            name: Cosine Precision@100
          - type: cosine_precision@150
            value: 0.2575
            name: Cosine Precision@150
          - type: cosine_precision@200
            value: 0.2250328947368421
            name: Cosine Precision@200
          - type: cosine_recall@1
            value: 0.010176836251443517
            name: Cosine Recall@1
          - type: cosine_recall@20
            value: 0.13155034702974863
            name: Cosine Recall@20
          - type: cosine_recall@50
            value: 0.2509109642167453
            name: Cosine Recall@50
          - type: cosine_recall@100
            value: 0.38660292276765434
            name: Cosine Recall@100
          - type: cosine_recall@150
            value: 0.48222911118709905
            name: Cosine Recall@150
          - type: cosine_recall@200
            value: 0.555593320773802
            name: Cosine Recall@200
          - type: cosine_ndcg@1
            value: 0.7203947368421053
            name: Cosine Ndcg@1
          - type: cosine_ndcg@20
            value: 0.5307474055663989
            name: Cosine Ndcg@20
          - type: cosine_ndcg@50
            value: 0.44447520594610196
            name: Cosine Ndcg@50
          - type: cosine_ndcg@100
            value: 0.434845358559754
            name: Cosine Ndcg@100
          - type: cosine_ndcg@150
            value: 0.47821684318798985
            name: Cosine Ndcg@150
          - type: cosine_ndcg@200
            value: 0.5217544441967603
            name: Cosine Ndcg@200
          - type: cosine_mrr@1
            value: 0.7203947368421053
            name: Cosine Mrr@1
          - type: cosine_mrr@20
            value: 0.8361781096010973
            name: Cosine Mrr@20
          - type: cosine_mrr@50
            value: 0.8361781096010973
            name: Cosine Mrr@50
          - type: cosine_mrr@100
            value: 0.8361781096010973
            name: Cosine Mrr@100
          - type: cosine_mrr@150
            value: 0.8361781096010973
            name: Cosine Mrr@150
          - type: cosine_mrr@200
            value: 0.8361781096010973
            name: Cosine Mrr@200
          - type: cosine_map@1
            value: 0.7203947368421053
            name: Cosine Map@1
          - type: cosine_map@20
            value: 0.3349546102812917
            name: Cosine Map@20
          - type: cosine_map@50
            value: 0.2333868330865278
            name: Cosine Map@50
          - type: cosine_map@100
            value: 0.20565538628439212
            name: Cosine Map@100
          - type: cosine_map@150
            value: 0.2210663674628728
            name: Cosine Map@150
          - type: cosine_map@200
            value: 0.2387037675984284
            name: Cosine Map@200
          - type: cosine_map@500
            value: 0.28801647424274696
            name: Cosine Map@500

Job-Skill matching fintuned BAAI/bge-large-en-v1.5 (v2)

Top performing model on TalentCLEF 2025 Task B. Use it for job title <-> skill set matching

Model Details

Model Description

• Model Type: Sentence Transformer
• Base model: BAAI/bge-large-en-v1.5
• Maximum Sequence Length: 512 tokens
• Output Dimensionality: 1024 dimensions
• Similarity Function: Cosine Similarity

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': 512, 'do_lower_case': True}) with Transformer model: BertModel 
  (1): Pooling({'word_embedding_dimension': 1024, 'pooling_mode_cls_token': True, 'pooling_mode_mean_tokens': False, '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("pj-mathematician/JobSkillBGE-large-en-v1.5-v2")
# Run inference
sentences = [
    "An insulation supervisor, regardless of the specific type of insulation material or installation area, requires strong project management skills, knowledge of building codes and safety regulations, and expertise in insulation techniques to oversee the installation process effectively and ensure quality standards are met.\n['insulation supervisor', 'supervisor of installation of insulating materials', 'supervisor of insulation materials installation', 'supervisor of installation of insulation', 'solid wall insulation installation supervisor', 'insulation installers supervisor', 'cavity wall insulation installation supervisor', 'loft insulation installation supervisor']",
    "The skill of installing insulation material is primarily required by job roles such as insulation workers, HVAC technicians, and construction specialists, who are responsible for improving energy efficiency and thermal comfort in buildings by correctly fitting and fixing insulation materials in various structures.\n['install insulation material', 'insulate structure', 'fix insulation', 'insulation material installation', 'installation of insulation material', 'fitting insulation', 'insulating structure', 'installing insulation material', 'fixing insulation', 'fit insulation']",
    "Job roles such as Food Safety Inspector, Public Health Officer, and Environmental Health Specialist require the skill of taking action on food safety violations to ensure compliance with health regulations and maintain public safety standards.\n['take action on food safety violations', 'invoke action on food safety violations', 'agree action on food safety violations', 'pursue action on food safety violations', 'determine action on food safety violations']",
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 1024]

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

Evaluation

Metrics

Information Retrieval

• Dataset: full_en
• Evaluated with InformationRetrievalEvaluator

Metric	Value
cosine_accuracy@1	0.7204
cosine_accuracy@20	1.0
cosine_accuracy@50	1.0
cosine_accuracy@100	1.0
cosine_accuracy@150	1.0
cosine_accuracy@200	1.0
cosine_precision@1	0.7204
cosine_precision@20	0.4916
cosine_precision@50	0.387
cosine_precision@100	0.3046
cosine_precision@150	0.2575
cosine_precision@200	0.225
cosine_recall@1	0.0102
cosine_recall@20	0.1316
cosine_recall@50	0.2509
cosine_recall@100	0.3866
cosine_recall@150	0.4822
cosine_recall@200	0.5556
cosine_ndcg@1	0.7204
cosine_ndcg@20	0.5307
cosine_ndcg@50	0.4445
cosine_ndcg@100	0.4348
cosine_ndcg@150	0.4782
cosine_ndcg@200	0.5218
cosine_mrr@1	0.7204
cosine_mrr@20	0.8362
cosine_mrr@50	0.8362
cosine_mrr@100	0.8362
cosine_mrr@150	0.8362
cosine_mrr@200	0.8362
cosine_map@1	0.7204
cosine_map@20	0.335
cosine_map@50	0.2334
cosine_map@100	0.2057
cosine_map@150	0.2211
cosine_map@200	0.2387
cosine_map@500	0.288

Training Details

Training Dataset

Unnamed Dataset

• Size: 114,699 training samples
• Columns: anchor and positive
• Approximate statistics based on the first 1000 samples:

  	anchor	positive
  type	string	string
  details	min: 78 tokens mean: 144.94 tokens max: 354 tokens	min: 51 tokens mean: 114.13 tokens max: 274 tokens

• Samples:

  anchor	positive
  A technical director or any of its synonyms requires a strong blend of technical expertise and leadership skills, including the ability to oversee technical operations, manage teams, and ensure the successful execution of technical projects while maintaining operational efficiency and innovation. ['technical director', 'technical and operations director', 'head of technical', 'director of technical arts', 'head of technical department', 'technical supervisor', 'technical manager']	Job roles that require promoting health and safety include occupational health and safety specialists, safety managers, and public health educators, all of whom work to ensure safe and healthy environments in workplaces and communities. ['promote health and safety', 'promote importance of health and safety', 'promoting health and safety', 'advertise health and safety']
  A technical director or any of its synonyms requires a strong blend of technical expertise and leadership skills, including the ability to oversee technical operations, manage teams, and ensure the successful execution of technical projects while maintaining operational efficiency and innovation. ['technical director', 'technical and operations director', 'head of technical', 'director of technical arts', 'head of technical department', 'technical supervisor', 'technical manager']	Job roles that require organizing rehearsals include directors, choreographers, and conductors in theater, dance, and music ensembles, who must efficiently plan and schedule practice sessions to prepare performers for a successful final performance. ['organise rehearsals', 'organise rehearsal', 'organize rehearsals', 'plan rehearsals', 'arrange rehearsals', 'organising rehearsals', 'schedule rehearsals']
  A technical director or any of its synonyms requires a strong blend of technical expertise and leadership skills, including the ability to oversee technical operations, manage teams, and ensure the successful execution of technical projects while maintaining operational efficiency and innovation. ['technical director', 'technical and operations director', 'head of technical', 'director of technical arts', 'head of technical department', 'technical supervisor', 'technical manager']	Job roles such as Health and Safety Managers, Environmental Health Officers, and Risk Management Specialists often require the skill of negotiating health and safety issues with third parties to ensure compliance and protection standards are met across different organizations and sites. ['negotiate health and safety issues with third parties', 'agree with third parties on health and safety', 'negotiate issues on health and safety with third parties', 'negotiate with third parties on health and safety issues', 'negotiate health and safety matters with third parties']

• Loss: CachedGISTEmbedLoss with these parameters:

  {'guide': SentenceTransformer(
    (0): Transformer({'max_seq_length': 128, 'do_lower_case': False}) with Transformer model: 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()
  ), 'temperature': 0.01, 'mini_batch_size': 32, 'margin_strategy': 'absolute', 'margin': 0.0}
  

Training Hyperparameters

Non-Default Hyperparameters

• eval_strategy: steps
• per_device_train_batch_size: 128
• per_device_eval_batch_size: 128
• gradient_accumulation_steps: 2
• num_train_epochs: 5
• warmup_ratio: 0.05
• log_on_each_node: False
• fp16: True
• dataloader_num_workers: 4
• ddp_find_unused_parameters: True
• batch_sampler: no_duplicates

All Hyperparameters

Click to expand

• overwrite_output_dir: False
• do_predict: False
• eval_strategy: steps
• prediction_loss_only: True
• per_device_train_batch_size: 128
• per_device_eval_batch_size: 128
• per_gpu_train_batch_size: None
• per_gpu_eval_batch_size: None
• gradient_accumulation_steps: 2
• eval_accumulation_steps: None
• torch_empty_cache_steps: None
• learning_rate: 5e-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: 5
• 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: False
• 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: True
• dataloader_num_workers: 4
• 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}
• tp_size: 0
• 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: True
• 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
• 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	full_en_cosine_ndcg@200
-1	-1	-	0.4795
0.0022	1	10.6462	-
0.2232	100	4.5115	-
0.4464	200	2.9237	0.5255
0.6696	300	2.5327	-
0.8929	400	2.3451	0.5305
1.1161	500	1.9882	-
1.3393	600	1.7738	0.5240
1.5625	700	1.7365	-
1.7857	800	1.6932	0.5251
2.0089	900	1.6184	-
2.2321	1000	1.285	0.5254
2.4554	1100	1.2651	-
2.6786	1200	1.2739	0.5238
2.9018	1300	1.2625	-
3.125	1400	1.0726	0.5251
3.3482	1500	0.9606	-
3.5714	1600	0.9594	0.5214
3.7946	1700	0.954	-
4.0179	1800	0.9264	0.5239
4.2411	1900	0.7486	-
4.4643	2000	0.7424	0.5218
4.6875	2100	0.7127	-
4.9107	2200	0.7129	0.5218

Framework Versions

• Python: 3.11.11
• Sentence Transformers: 4.1.0
• Transformers: 4.51.2
• PyTorch: 2.6.0+cu124
• Accelerate: 1.6.0
• Datasets: 3.5.0
• Tokenizers: 0.21.1

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