SentenceTransformer(
(0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: MPNetModel
(1): Pooling({'word_embedding_dimension': 768, '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})
)pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("Marco127/Base_T")
5# Run inference
6sentences = [
7 '\nGuests are responsible for damages caused to hotel property according to the valid legal\nprescriptions of Hungary.',
8 '\nGuests are responsible for damages caused to hotel property according to the valid legal\nprescriptions of Hungary.',
9 '\nWe shall be happy to listen to any suggestions for improvement of the accommodation\nand catering services in the hotel. In case of any complaints we shall purposefully arrange\nthe rectification of any insufficiencies.',
10]
11embeddings = model.encode(sentences)
12print(embeddings.shape)
13# [3, 768]
14
15# Get the similarity scores for the embeddings
16similarities = model.similarity(embeddings, embeddings)
17print(similarities.shape)
18# [3, 3]BinaryClassificationEvaluator| Metric | Value |
|---|---|
| dot_accuracy | 0.655 |
| dot_accuracy_threshold | 48.3617 |
| dot_f1 | 0.5143 |
| dot_f1_threshold | 40.0116 |
| dot_precision | 0.36 |
| dot_recall | 0.9 |
| dot_ap | 0.3571 |
| dot_mcc | 0.0388 |
sentence1, sentence2, and label| sentence1 | sentence2 | label | |
|---|---|---|---|
| type | string | string | int |
| details |
|
|
|
| sentence1 | sentence2 | label |
|---|---|---|
If a guest fails to vacate[object Object]the room within the designated time, reception shall charge this guest for the following[object Object]night's accommodation fee. | If a guest fails to vacate[object Object]the room within the designated time, reception shall charge this guest for the following[object Object]night's accommodation fee. | 0 |
If you do not want someone to enter[object Object]your room, please hang the "do not disturb” card on your room’s outside door handle. It can[object Object]be found in the entrance area of your room. | If you do not want someone to enter[object Object]your room, please hang the "do not disturb” card on your room’s outside door handle. It can[object Object]be found in the entrance area of your room. | 0 |
[object Object]Owners are responsible for ensuring that animals are kept quiet between the[object Object]hours of 10:00 pm and 06:00 am. In the case of failure to abide by this[object Object]regulation the guest may be asked to leave the hotel without a refund of the[object Object]price of the night's accommodation. | [object Object]Owners are responsible for ensuring that animals are kept quiet between the[object Object]hours of 10:00 pm and 06:00 am. In the case of failure to abide by this[object Object]regulation the guest may be asked to leave the hotel without a refund of the[object Object]price of the night's accommodation. | 0 |
MultipleNegativesRankingLoss with these parameters:
1{
2 "scale": 20.0,
3 "similarity_fct": "cos_sim"
4}sentence1, sentence2, and label| sentence1 | sentence2 | label | |
|---|---|---|---|
| type | string | string | int |
| details |
|
|
|
| sentence1 | sentence2 | label |
|---|---|---|
[object Object]We shall be happy to listen to any suggestions for improvement of the accommodation[object Object]and catering services in the hotel. In case of any complaints we shall purposefully arrange[object Object]the rectification of any insufficiencies. | [object Object]We shall be happy to listen to any suggestions for improvement of the accommodation[object Object]and catering services in the hotel. In case of any complaints we shall purposefully arrange[object Object]the rectification of any insufficiencies. | 0 |
[object Object]Between the hours of 10:00 pm and 06:00 am guests are obliged to maintain low noise[object Object]levels. | [object Object]Between the hours of 10:00 pm and 06:00 am guests are obliged to maintain low noise[object Object]levels. | 0 |
[object Object]The hotel’s inner courtyard parking facility may be used only upon availability of parking[object Object]slots. Slots marked as ’Private’ are to be left free for their owners. For parking fees please[object Object]consult the reception or see the website of the hotel. | [object Object]The hotel’s inner courtyard parking facility may be used only upon availability of parking[object Object]slots. Slots marked as ’Private’ are to be left free for their owners. For parking fees please[object Object]consult the reception or see the website of the hotel. | 1 |
MultipleNegativesRankingLoss with these parameters:
1{
2 "scale": 20.0,
3 "similarity_fct": "cos_sim"
4}eval_strategy: stepsper_device_train_batch_size: 16per_device_eval_batch_size: 16learning_rate: 2e-05num_train_epochs: 5warmup_ratio: 0.1fp16: Truebatch_sampler: no_duplicatesoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: stepsprediction_loss_only: Trueper_device_train_batch_size: 16per_device_eval_batch_size: 16per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 1eval_accumulation_steps: Nonetorch_empty_cache_steps: Nonelearning_rate: 2e-05weight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08max_grad_norm: 1.0num_train_epochs: 5max_steps: -1lr_scheduler_type: linearlr_scheduler_kwargs: {}warmup_ratio: 0.1warmup_steps: 0log_level: passivelog_level_replica: warninglog_on_each_node: Truelogging_nan_inf_filter: Truesave_safetensors: Truesave_on_each_node: Falsesave_only_model: Falserestore_callback_states_from_checkpoint: Falseno_cuda: Falseuse_cpu: Falseuse_mps_device: Falseseed: 42data_seed: Nonejit_mode_eval: Falseuse_ipex: Falsebf16: Falsefp16: Truefp16_opt_level: O1half_precision_backend: autobf16_full_eval: Falsefp16_full_eval: Falsetf32: Nonelocal_rank: 0ddp_backend: Nonetpu_num_cores: Nonetpu_metrics_debug: Falsedebug: []dataloader_drop_last: Falsedataloader_num_workers: 0dataloader_prefetch_factor: Nonepast_index: -1disable_tqdm: Falseremove_unused_columns: Truelabel_names: Noneload_best_model_at_end: Falseignore_data_skip: Falsefsdp: []fsdp_min_num_params: 0fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}fsdp_transformer_layer_cls_to_wrap: Noneaccelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}deepspeed: Nonelabel_smoothing_factor: 0.0optim: adamw_torchoptim_args: Noneadafactor: Falsegroup_by_length: Falselength_column_name: lengthddp_find_unused_parameters: Noneddp_bucket_cap_mb: Noneddp_broadcast_buffers: Falsedataloader_pin_memory: Truedataloader_persistent_workers: Falseskip_memory_metrics: Trueuse_legacy_prediction_loop: Falsepush_to_hub: Falseresume_from_checkpoint: Nonehub_model_id: Nonehub_strategy: every_savehub_private_repo: Nonehub_always_push: Falsegradient_checkpointing: Falsegradient_checkpointing_kwargs: Noneinclude_inputs_for_metrics: Falseinclude_for_metrics: []eval_do_concat_batches: Truefp16_backend: autopush_to_hub_model_id: Nonepush_to_hub_organization: Nonemp_parameters:auto_find_batch_size: Falsefull_determinism: Falsetorchdynamo: Noneray_scope: lastddp_timeout: 1800torch_compile: Falsetorch_compile_backend: Nonetorch_compile_mode: Nonedispatch_batches: Nonesplit_batches: Noneinclude_tokens_per_second: Falseinclude_num_input_tokens_seen: Falseneftune_noise_alpha: Noneoptim_target_modules: Nonebatch_eval_metrics: Falseeval_on_start: Falseuse_liger_kernel: Falseeval_use_gather_object: Falseaverage_tokens_across_devices: Falseprompts: Nonebatch_sampler: no_duplicatesmulti_dataset_batch_sampler: proportional| Epoch | Step | Training Loss | Validation Loss | dot_ap |
|---|---|---|---|---|
| -1 | -1 | - | - | 0.3571 |
| 2.2791 | 100 | 0.0011 | 0.0000 | - |
| 4.5581 | 200 | 0.0 | 0.0000 | - |
1@inproceedings{reimers-2019-sentence-bert,
2 title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
3 author = "Reimers, Nils and Gurevych, Iryna",
4 booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
5 month = "11",
6 year = "2019",
7 publisher = "Association for Computational Linguistics",
8 url = "https://arxiv.org/abs/1908.10084",
9}1@misc{henderson2017efficient,
2 title={Efficient Natural Language Response Suggestion for Smart Reply},
3 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},
4 year={2017},
5 eprint={1705.00652},
6 archivePrefix={arXiv},
7 primaryClass={cs.CL}
8}