Views
No views yet
SentenceTransformer(
(0): Transformer({'transformer_task': 'feature-extraction', 'modality_config': {'text': {'method': 'get_text_features', 'method_output_name': None}, 'audio': {'method': 'get_audio_features', 'method_output_name': None}}, 'module_output_name': 'sentence_embedding', 'architecture': 'ClapModel'})
)pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("tomaarsen/clap-htsat-fused-librispeech")
5# Run inference
6sentences = [
7 'THERE ARE NATURES TOO TO WHOSE SENSE OF JUSTICE THE PRICE EXACTED LOOMS UP MONSTROUSLY ENORMOUS ODIOUS OPPRESSIVE WORRYING HUMILIATING EXTORTIONATE INTOLERABLE THOSE ARE THE FANATICS',
8 'HE BEGAN TO WISH THAT HE HAD COMPROMISED IN SOME WAY OR OTHER THAT HE HAD SENT THE MONEY PERHAPS HE COULD DO IT UP HERE',
9 'HERE THE HOLY PRELATE OF FERNS MET HIM AND RELATED A VISION IN WHICH HE HAD BEEN INSTRUCTED TO DEMAND THE ABOLITION OF THE IMPOST',
10]
11embeddings = model.encode(sentences)
12print(embeddings.shape)
13# [3, 1024]
14
15# Get the similarity scores for the embeddings
16similarities = model.similarity(embeddings, embeddings)
17print(similarities)
18# tensor([[ 1.0000, -0.4742, -0.2719],
19# [-0.4742, 1.0000, 0.8206],
20# [-0.2719, 0.8206, 1.0000]])librispeech-eval and librispeech-testInformationRetrievalEvaluator| Metric | librispeech-eval | librispeech-test |
|---|---|---|
| cosine_accuracy@1 | 0.108 | 0.151 |
| cosine_accuracy@3 | 0.196 | 0.288 |
| cosine_accuracy@5 | 0.272 | 0.371 |
| cosine_accuracy@10 | 0.438 | 0.518 |
| cosine_precision@1 | 0.108 | 0.151 |
| cosine_precision@3 | 0.0653 | 0.096 |
| cosine_precision@5 | 0.0544 | 0.0742 |
| cosine_precision@10 | 0.0438 | 0.0518 |
| cosine_recall@1 | 0.108 | 0.151 |
| cosine_recall@3 | 0.196 | 0.288 |
| cosine_recall@5 | 0.272 | 0.371 |
| cosine_recall@10 | 0.438 | 0.518 |
| cosine_ndcg@10 | 0.2432 | 0.3132 |
| cosine_mrr@10 | 0.1849 | 0.2505 |
| cosine_map@100 | 0.206 | 0.2694 |
audio and text| audio | text | |
|---|---|---|
| type | dict | string |
| details |
|
| audio | text |
|---|---|
{'path': '374-180298-0000.flac', 'array': array([ 6.92203816e-04, 8.04404495e-04, 8.03834875e-04, ...,[object Object] -3.02505396e-05, -6.59527450e-06, 1.11444592e-06]), 'sampling_rate': 48000} | CHAPTER SIXTEEN I MIGHT HAVE TOLD YOU OF THE BEGINNING OF THIS LIAISON IN A FEW LINES BUT I WANTED YOU TO SEE EVERY STEP BY WHICH WE CAME I TO AGREE TO WHATEVER MARGUERITE WISHED |
{'path': '374-180298-0001.flac', 'array': array([-9.33515839e-05, -1.25754057e-04, -1.44482241e-04, ...,[object Object] -2.66165182e-04, -2.03228556e-04, -1.03404833e-04]), 'sampling_rate': 48000} | MARGUERITE TO BE UNABLE TO LIVE APART FROM ME IT WAS THE DAY AFTER THE EVENING WHEN SHE CAME TO SEE ME THAT I SENT HER MANON LESCAUT FROM THAT TIME SEEING THAT I COULD NOT CHANGE MY MISTRESS'S LIFE I CHANGED MY OWN |
{'path': '374-180298-0002.flac', 'array': array([-2.47883319e-04, -2.91854434e-04, -2.82971043e-04, ...,[object Object] -1.43931946e-04, -1.17829914e-04, -6.32331648e-05]), 'sampling_rate': 48000} | I WISHED ABOVE ALL NOT TO LEAVE MYSELF TIME TO THINK OVER THE POSITION I HAD ACCEPTED FOR IN SPITE OF MYSELF IT WAS A GREAT DISTRESS TO ME THUS MY LIFE GENERALLY SO CALM |
MultipleNegativesSymmetricRankingLoss with these parameters:
1{
2 "scale": 20.0,
3 "similarity_fct": "cos_sim",
4 "gather_across_devices": false
5}audio and text| audio | text | |
|---|---|---|
| type | dict | string |
| details |
|
| audio | text |
|---|---|
{'path': '2277-149896-0000.flac', 'array': array([ 0.00179741, 0.00170625, 0.00120927, ..., -0.00144462,[object Object] -0.00102732, -0.00048062]), 'sampling_rate': 48000} | HE WAS IN A FEVERED STATE OF MIND OWING TO THE BLIGHT HIS WIFE'S ACTION THREATENED TO CAST UPON HIS ENTIRE FUTURE |
{'path': '2277-149896-0001.flac', 'array': array([ 0.00111104, 0.00081758, 0.00021103, ..., -0.00138193,[object Object] -0.0009173 , -0.00041702]), 'sampling_rate': 48000} | HE WOULD HAVE TO PAY HER THE MONEY WHICH SHE WOULD NOW REGULARLY DEMAND OR THERE WOULD BE TROUBLE IT DID NOT MATTER WHAT HE DID |
{'path': '2277-149896-0002.flac', 'array': array([0.00080266, 0.00088462, 0.00083408, ..., 0.00105488, 0.00083673,[object Object] 0.00043296]), 'sampling_rate': 48000} | HURSTWOOD WALKED THE FLOOR MENTALLY ARRANGING THE CHIEF POINTS OF HIS SITUATION |
MultipleNegativesSymmetricRankingLoss with these parameters:
1{
2 "scale": 20.0,
3 "similarity_fct": "cos_sim",
4 "gather_across_devices": false
5}eval_strategy: stepsper_device_train_batch_size: 16per_device_eval_batch_size: 16learning_rate: 2e-05num_train_epochs: 1warmup_ratio: 0.1bf16: Truebatch_sampler: no_duplicatesoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: stepsprediction_loss_only: Trueper_device_train_batch_size: 16per_device_eval_batch_size: 16gradient_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: 1max_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: Falseuse_cpu: Falseseed: 42data_seed: Nonejit_mode_eval: Falsebf16: Truefp16: Falsehalf_precision_backend: Nonebf16_full_eval: Falsefp16_full_eval: Falsetf32: Nonelocal_rank: 0ddp_backend: Nonetpu_num_cores: Nonedebug: []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_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}accelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}parallelism_config: Nonedeepspeed: Nonelabel_smoothing_factor: 0.0optim: adamw_torch_fusedoptim_args: Nonegroup_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: Falsehub_revision: Nonegradient_checkpointing: Falsegradient_checkpointing_kwargs: Noneinclude_for_metrics: []eval_do_concat_batches: Truemp_parameters:auto_find_batch_size: Falsefull_determinism: Falseray_scope: lastddp_timeout: 1800torch_compile: Falsetorch_compile_backend: Nonetorch_compile_mode: Noneinclude_tokens_per_second: Falseinclude_num_input_tokens_seen: noneftune_noise_alpha: Noneoptim_target_modules: Nonebatch_eval_metrics: Falseeval_on_start: Falseuse_liger_kernel: Falseliger_kernel_config: Noneeval_use_gather_object: Falseaverage_tokens_across_devices: Trueprompts: Nonebatch_sampler: no_duplicatesmulti_dataset_batch_sampler: proportionalrouter_mapping: {}learning_rate_mapping: {}| Epoch | Step | Training Loss | Validation Loss | librispeech-eval_cosine_ndcg@10 | librispeech-test_cosine_ndcg@10 |
|---|---|---|---|---|---|
| -1 | -1 | - | - | 0.0114 | - |
| 0.0100 | 83 | 3.5908 | - | - | - |
| 0.0200 | 166 | 2.5371 | - | - | - |
| 0.0301 | 249 | 2.1799 | - | - | - |
| 0.0401 | 332 | 2.0415 | - | - | - |
| 0.0501 | 415 | 1.9394 | - | - | - |
| 0.0601 | 498 | 1.8167 | - | - | - |
| 0.0701 | 581 | 1.7589 | - | - | - |
| 0.0801 | 664 | 1.7262 | - | - | - |
| 0.0902 | 747 | 1.7585 | - | - | - |
| 0.1001 | 829 | - | 1.5991 | 0.0335 | - |
| 0.1002 | 830 | 1.7521 | - | - | - |
| 0.1102 | 913 | 1.6822 | - | - | - |
| 0.1202 | 996 | 1.6176 | - | - | - |
| 0.1302 | 1079 | 1.6391 | - | - | - |
| 0.1403 | 1162 | 1.6931 | - | - | - |
| 0.1503 | 1245 | 1.4626 | - | - | - |
| 0.1603 | 1328 | 1.4305 | - | - | - |
| 0.1703 | 1411 | 1.4998 | - | - | - |
| 0.1803 | 1494 | 1.4073 | - | - | - |
| 0.1903 | 1577 | 1.3843 | - | - | - |
| 0.2001 | 1658 | - | 1.2227 | 0.0925 | - |
| 0.2004 | 1660 | 1.3371 | - | - | - |
| 0.2104 | 1743 | 1.3908 | - | - | - |
| 0.2204 | 1826 | 1.2835 | - | - | - |
| 0.2304 | 1909 | 1.3203 | - | - | - |
| 0.2404 | 1992 | 1.2549 | - | - | - |
| 0.2505 | 2075 | 1.2384 | - | - | - |
| 0.2605 | 2158 | 1.2189 | - | - | - |
| 0.2705 | 2241 | 1.1658 | - | - | - |
| 0.2805 | 2324 | 1.1771 | - | - | - |
| 0.2905 | 2407 | 1.2068 | - | - | - |
| 0.3002 | 2487 | - | 1.0471 | 0.1318 | - |
| 0.3005 | 2490 | 1.1708 | - | - | - |
| 0.3106 | 2573 | 1.1389 | - | - | - |
| 0.3206 | 2656 | 1.0786 | - | - | - |
| 0.3306 | 2739 | 1.0792 | - | - | - |
| 0.3406 | 2822 | 1.0562 | - | - | - |
| 0.3506 | 2905 | 0.98 | - | - | - |
| 0.3607 | 2988 | 1.1153 | - | - | - |
| 0.3707 | 3071 | 0.9987 | - | - | - |
| 0.3807 | 3154 | 1.0002 | - | - | - |
| 0.3907 | 3237 | 1.0017 | - | - | - |
| 0.4002 | 3316 | - | 0.8901 | 0.1589 | - |
| 0.4007 | 3320 | 0.9364 | - | - | - |
| 0.4107 | 3403 | 0.9394 | - | - | - |
| 0.4208 | 3486 | 0.9459 | - | - | - |
| 0.4308 | 3569 | 0.9604 | - | - | - |
| 0.4408 | 3652 | 0.9491 | - | - | - |
| 0.4508 | 3735 | 0.9295 | - | - | - |
| 0.4608 | 3818 | 0.9508 | - | - | - |
| 0.4709 | 3901 | 0.9122 | - | - | - |
| 0.4809 | 3984 | 0.8483 | - | - | - |
| 0.4909 | 4067 | 0.8443 | - | - | - |
| 0.5003 | 4145 | - | 0.7955 | 0.1908 | - |
| 0.5009 | 4150 | 0.8838 | - | - | - |
| 0.5109 | 4233 | 0.8367 | - | - | - |
| 0.5209 | 4316 | 0.8516 | - | - | - |
| 0.5310 | 4399 | 0.8112 | - | - | - |
| 0.5410 | 4482 | 0.8368 | - | - | - |
| 0.5510 | 4565 | 0.873 | - | - | - |
| 0.5610 | 4648 | 0.8156 | - | - | - |
| 0.5710 | 4731 | 0.8864 | - | - | - |
| 0.5811 | 4814 | 0.8278 | - | - | - |
| 0.5911 | 4897 | 0.8006 | - | - | - |
| 0.6004 | 4974 | - | 0.7649 | 0.1874 | - |
| 0.6011 | 4980 | 0.8199 | - | - | - |
| 0.6111 | 5063 | 0.7475 | - | - | - |
| 0.6211 | 5146 | 0.7345 | - | - | - |
| 0.6311 | 5229 | 0.7301 | - | - | - |
| 0.6412 | 5312 | 0.774 | - | - | - |
| 0.6512 | 5395 | 0.7391 | - | - | - |
| 0.6612 | 5478 | 0.6929 | - | - | - |
| 0.6712 | 5561 | 0.7218 | - | - | - |
| 0.6812 | 5644 | 0.7071 | - | - | - |
| 0.6912 | 5727 | 0.7024 | - | - | - |
| 0.7004 | 5803 | - | 0.6712 | 0.2419 | - |
| 0.7013 | 5810 | 0.6428 | - | - | - |
| 0.7113 | 5893 | 0.6719 | - | - | - |
| 0.7213 | 5976 | 0.6972 | - | - | - |
| 0.7313 | 6059 | 0.7043 | - | - | - |
| 0.7413 | 6142 | 0.663 | - | - | - |
| 0.7514 | 6225 | 0.6963 | - | - | - |
| 0.7614 | 6308 | 0.6591 | - | - | - |
| 0.7714 | 6391 | 0.6736 | - | - | - |
| 0.7814 | 6474 | 0.7033 | - | - | - |
| 0.7914 | 6557 | 0.6314 | - | - | - |
| 0.8005 | 6632 | - | 0.6806 | 0.2319 | - |
| 0.8014 | 6640 | 0.6508 | - | - | - |
| 0.8115 | 6723 | 0.6532 | - | - | - |
| 0.8215 | 6806 | 0.6788 | - | - | - |
| 0.8315 | 6889 | 0.6038 | - | - | - |
| 0.8415 | 6972 | 0.658 | - | - | - |
| 0.8515 | 7055 | 0.656 | - | - | - |
| 0.8616 | 7138 | 0.6533 | - | - | - |
| 0.8716 | 7221 | 0.601 | - | - | - |
| 0.8816 | 7304 | 0.6243 | - | - | - |
| 0.8916 | 7387 | 0.6315 | - | - | - |
| 0.9005 | 7461 | - | 0.6526 | 0.2432 | - |
| 0.9016 | 7470 | 0.5707 | - | - | - |
| 0.9116 | 7553 | 0.5778 | - | - | - |
| 0.9217 | 7636 | 0.5736 | - | - | - |
| 0.9317 | 7719 | 0.615 | - | - | - |
| 0.9417 | 7802 | 0.5756 | - | - | - |
| 0.9517 | 7885 | 0.5724 | - | - | - |
| 0.9617 | 7968 | 0.5678 | - | - | - |
| 0.9718 | 8051 | 0.5661 | - | - | - |
| 0.9818 | 8134 | 0.6162 | - | - | - |
| 0.9918 | 8217 | 0.5766 | - | - | - |
| -1 | -1 | - | - | - | 0.3132 |
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}