CoolFace
Modelpublic

stephenhib/all-mpnet-base-v2-patabs-1epoc-batch32-100000

sourceHugging Faceupdated 2y agoView on Hugging Face
0likes85downloads
Model Card

SentenceTransformer based on sentence-transformers/all-mpnet-base-v2

This is a sentence-transformers model finetuned from sentence-transformers/all-mpnet-base-v2 on the json 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 Type: Sentence Transformer
  • —Base model: sentence-transformers/all-mpnet-base-v2 <!-- at revision f1b1b820e405bb8644f5e8d9a3b98f9c9e0a3c58 -->
  • —Maximum Sequence Length: 384 tokens
  • —Output Dimensionality: 768 tokens
  • —Similarity Function: Cosine Similarity
  • —Training Dataset:
  • —json <!-- - Language: Unknown --> <!-- - License: Unknown -->

Model Sources

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 384, 'do_lower_case': False}) with Transformer model: MPNetModel 
  (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})
  (2): Normalize()
)

Usage

Direct Usage (Sentence Transformers)

First install the Sentence Transformers library:

bash
pip install -U sentence-transformers

Then you can load this model and run inference.

python
from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("stephenhib/all-mpnet-base-v2-patabs-1epoc-batch32-100000")
# Run inference
sentences = [
    '<p id="pa01" num="0001">An application apparatus (100) includes: an application needle (24) that applies, to a target, an application material having its viscosity changing under shear; a drive unit (90) that moves the application needle (24) up and down; and a controller (80) that controls the drive unit (90) to move the application needle such that shear is applied to the application material at a shear speed depending on a type of the application material and depending on a target application amount or a target application diameter.<img id="iaf01" file="imgaf001.tif" wi="78" he="56" img-content="drawing" img-format="tif"/></p>',
    'COATING APPARATUS AND COATING METHOD',
    'Electric motor',
]
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]

<!--

Direct Usage (Transformers)

<details><summary>Click to see the direct usage in Transformers</summary>

</details> -->

<!--

Downstream Usage (Sentence Transformers)

You can finetune this model on your own dataset.

<details><summary>Click to expand</summary>

</details> -->

<!--

Out-of-Scope Use

List how the model may foreseeably be misused and address what users ought not to do with the model. -->

Evaluation

Metrics

Information Retrieval
MetricValue
cosine_accuracy@10.592
cosine_accuracy@30.711
cosine_accuracy@50.751
cosine_accuracy@100.814
cosine_precision@10.592
cosine_precision@30.237
cosine_precision@50.1502
cosine_precision@100.0814
cosine_recall@10.592
cosine_recall@30.711
cosine_recall@50.751
cosine_recall@100.814
cosine_ndcg@100.6988
cosine_mrr@100.6625
cosine_map@1000.6665

<!--

Bias, Risks and Limitations

What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model. -->

<!--

Recommendations

What are recommendations with respect to the foreseeable issues? For example, filtering explicit content. -->

Training Details

Training Dataset

json
  • —Dataset: json
  • —Size: 807,656 training samples
  • —Columns: <code>positive</code> and <code>anchor</code>
  • —Approximate statistics based on the first 1000 samples: | | positive | anchor | |:--------|:-------------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------| | type | string | string | | details | <ul><li>min: 45 tokens</li><li>mean: 237.14 tokens</li><li>max: 384 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 12.34 tokens</li><li>max: 101 tokens</li></ul> |
  • —Samples: | positive | anchor | |:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------------------------------------------| | <code><p id="pa01" num="0001">The invention relates to an image fusion method and device, which includes: obtaining a first short-focus image and a first long-focus image acquired by a short-focus sensor and a long-focus sensor at the same time; according to the focal lengths of a short-focus lens and a long-focus lens, calculating a reduction coefficient corresponding to the first long-focus image when the sizes of the same target in the first long-focus image and the first short-focus image are matched; performing a reduction processing on the first long-focus image according to the reduction coefficient to obtain a second long-focus image; according to a relative angle of the current long-focus lens and short-focus lens, calculating a position of the second long-focus image in the first short-focus image when the positions of the same target in the second long-focus image and the first short-focus image are matched; and according to the position of the second long-focus image in the first short-focus image, covering the first short-focus image by the second long-focus image to obtain a fused image. According to embodiments of the present application, on the premise of considering both the monitoring range and the definition, the monitoring cost is reduced, and the monitoring efficiency is improved.<img id="iaf01" file="imgaf001.tif" wi="92" he="72" img-content="drawing" img-format="tif"/></p></code> | <code>IMAGE FUSION METHOD AND DEVICE</code> | | <code><p id="pa01" num="0001">The present invention discloses an <i>ex vivo</i> method for the diagnostic and/or prognostic assessment of the acute-on-chronic liver failure (ACLF) syndrome in a patient with a liver disorder characterized in that it comprises the steps of: (a) measuring a panel of metabolites related with acylcarnitines-sialic acid-acetylated amino acids and/or sugar alcohols and derivatives-tryptophan metabolism-catecholamines derivatives in a biological sample of said patient; and (b) comparing the level of said metabolites in the sample with the level of said metabolites in healthy patients; and wherein an increase of at least 1.2 times of the level of said metabolites is indicative of ACLF syndrome.</p></code> | <code>METHOD FOR THE DIAGNOSTIC AND/OR PROGNOSTIC ASSESSMENT OF ACUTE-ON-CHRONIC LIVER FAILURE SYNDROME IN PATIENTS WITH LIVER DISORDERS</code> | | <code><p id="pa01" num="0001">A valve housing receives a spool 34 and the spool has a regulating chamber 52 selectively communicating a supply line to a return line. The spool 34 is biased in one direction by a spring force and there is a second force biasing the spool in an opposed direction whith the second bias force being provided by a fluid pressure within a hydraulic system associated which the pressure regulating valve. The amount of communication between the supply port 111 and the return port 99 is regulated by a position of the spool 34 as the bias force from the fluid pressure change. Damper chambers are provided on opposed sides of the spool and serve to dampen a speed of movement of the spool and a supply line for supplying fluid into the damper chambers through check valves 44, 64. The supply line serves to assist in purging air outwardly of the damper chambers.<br><img id="iaf01" file="imgaf001.tif" wi="142" he="100" img-content="drawing" img-format="tif"/></p></code> | <code>Air purging pressure regulating valve</code> |
  • —Loss: <code>MultipleNegativesRankingLoss</code> with these parameters:
json
  {
      "scale": 20.0,
      "similarity_fct": "cos_sim"
  }

Training Hyperparameters

Non-Default Hyperparameters
  • —eval_strategy: steps
  • —per_device_train_batch_size: 4
  • —per_device_eval_batch_size: 2
  • —learning_rate: 2e-05
  • —num_train_epochs: 1
  • —warmup_ratio: 0.1
  • —bf16: True
  • —batch_sampler: no_duplicates
All Hyperparameters

<details><summary>Click to expand</summary>

  • —overwrite_output_dir: False
  • —do_predict: False
  • —eval_strategy: steps
  • —prediction_loss_only: True
  • —per_device_train_batch_size: 4
  • —per_device_eval_batch_size: 2
  • —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: 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: {'minnumparams': 0, 'xla': False, 'xlafsdpv2': False, 'xlafsdpgrad_ckpt': False}
  • —fsdp_transformer_layer_cls_to_wrap: None
  • —accelerator_config: {'splitbatches': False, 'dispatchbatches': None, 'evenbatches': True, 'useseedablesampler': True, 'nonblocking': False, 'gradientaccumulationkwargs': 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: False
  • —hub_always_push: False
  • —gradient_checkpointing: False
  • —gradient_checkpointing_kwargs: None
  • —include_inputs_for_metrics: False
  • —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
  • —batch_sampler: no_duplicates
  • —multi_dataset_batch_sampler: proportional

</details>

Training Logs

EpochStepTraining Losssentence-transformers/all-mpnet-base-v2_cosine_map@100
0.0321000.14330.6217
0.0642000.09530.6447
0.0963000.10840.6612
0.1284000.08170.6546
0.165000.07680.6512
0.1926000.07790.6466
0.2247000.07090.6594
0.2568000.08130.6441
0.2889000.05970.6454
0.3210000.07440.6496
0.35211000.06690.6608
0.38412000.06570.6566
0.41613000.04890.6660
0.44814000.06430.6597
0.4815000.05930.6587
0.51216000.05980.6613
0.54417000.07370.6570
0.57618000.06610.6655
0.60819000.04990.6613
0.6420000.06410.6616
0.67221000.06790.6662
0.70422000.05210.6715
0.73623000.05690.6651
0.76824000.05070.6679
0.825000.04050.6678
0.83226000.05480.6690
0.86427000.04030.6692
0.89628000.06130.6649
0.92829000.04850.6673
0.9630000.04950.6674
0.99231000.05460.6665

Framework Versions

  • —Python: 3.11.9
  • —Sentence Transformers: 3.2.1
  • —Transformers: 4.45.2
  • —PyTorch: 2.3.1.post300
  • —Accelerate: 1.0.1
  • —Datasets: 3.0.1
  • —Tokenizers: 0.20.1

Citation

BibTeX

Sentence Transformers
bibtex
@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",
}
MultipleNegativesRankingLoss
bibtex
@misc{henderson2017efficient,
    title={Efficient Natural Language Response Suggestion for Smart Reply},
    author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
    year={2017},
    eprint={1705.00652},
    archivePrefix={arXiv},
    primaryClass={cs.CL}
}

<!--

Glossary

Clearly define terms in order to be accessible across audiences. -->

<!--

Model Card Authors

Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction. -->

<!--

Model Card Contact

Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors. -->