Views
No views yet
SentenceTransformer(
(transformer): Transformer(
(auto_model): XLMRobertaLoRA(
(roberta): XLMRobertaModel(
(embeddings): XLMRobertaEmbeddings(
(word_embeddings): ParametrizedEmbedding(
250002, 1024, padding_idx=1
(parametrizations): ModuleDict(
(weight): ParametrizationList(
(0): LoRAParametrization()
)
)
)
(token_type_embeddings): ParametrizedEmbedding(
1, 1024
(parametrizations): ModuleDict(
(weight): ParametrizationList(
(0): LoRAParametrization()
)
)
)
)
(emb_drop): Dropout(p=0.1, inplace=False)
(emb_ln): LayerNorm((1024,), eps=1e-05, elementwise_affine=True)
(encoder): XLMRobertaEncoder(
(layers): ModuleList(
(0-23): 24 x Block(
(mixer): MHA(
(rotary_emb): RotaryEmbedding()
(Wqkv): ParametrizedLinearResidual(
in_features=1024, out_features=3072, bias=True
(parametrizations): ModuleDict(
(weight): ParametrizationList(
(0): LoRAParametrization()
)
)
)
(inner_attn): FlashSelfAttention(
(drop): Dropout(p=0.1, inplace=False)
)
(inner_cross_attn): FlashCrossAttention(
(drop): Dropout(p=0.1, inplace=False)
)
(out_proj): ParametrizedLinear(
in_features=1024, out_features=1024, bias=True
(parametrizations): ModuleDict(
(weight): ParametrizationList(
(0): LoRAParametrization()
)
)
)
)
(dropout1): Dropout(p=0.1, inplace=False)
(drop_path1): StochasticDepth(p=0.0, mode=row)
(norm1): LayerNorm((1024,), eps=1e-05, elementwise_affine=True)
(mlp): Mlp(
(fc1): ParametrizedLinear(
in_features=1024, out_features=4096, bias=True
(parametrizations): ModuleDict(
(weight): ParametrizationList(
(0): LoRAParametrization()
)
)
)
(fc2): ParametrizedLinear(
in_features=4096, out_features=1024, bias=True
(parametrizations): ModuleDict(
(weight): ParametrizationList(
(0): LoRAParametrization()
)
)
)
)
(dropout2): Dropout(p=0.1, inplace=False)
(drop_path2): StochasticDepth(p=0.0, mode=row)
(norm2): LayerNorm((1024,), eps=1e-05, elementwise_affine=True)
)
)
)
(pooler): XLMRobertaPooler(
(dense): ParametrizedLinear(
in_features=1024, out_features=1024, bias=True
(parametrizations): ModuleDict(
(weight): ParametrizationList(
(0): LoRAParametrization()
)
)
)
(activation): Tanh()
)
)
)
)
(pooler): Pooling({'word_embedding_dimension': 1024, '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})
(normalizer): Normalize()
)pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("damon6/de_shop_api_v3_jina-embeddings-v3-base-finetuned")
5# Run inference
6sentences = [
7 'DIGITUS Mini GBIC SFP Modul 10G Leistung',
8 'Die DIGITUS 10G Mini GBIC SFP Transceiver Module bieten hohe Qualität und Zuverlässigkeit.',
9 'eine 325 mm lange GPU',
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.shape)
18# [3, 3]dim_1024InformationRetrievalEvaluator with these parameters:
1{
2 "truncate_dim": 1024
3}| Metric | Value |
|---|---|
| cosine_accuracy@1 | 0.5348 |
| cosine_accuracy@3 | 0.7476 |
| cosine_accuracy@5 | 0.8063 |
| cosine_accuracy@10 | 0.8881 |
| cosine_precision@1 | 0.5348 |
| cosine_precision@3 | 0.2492 |
| cosine_precision@5 | 0.1613 |
| cosine_precision@10 | 0.0888 |
| cosine_recall@1 | 0.5348 |
| cosine_recall@3 | 0.7476 |
| cosine_recall@5 | 0.8063 |
| cosine_recall@10 | 0.8881 |
| cosine_ndcg@10 | 0.7126 |
| cosine_mrr@10 | 0.6564 |
| cosine_map@100 | 0.6597 |
dim_768InformationRetrievalEvaluator with these parameters:
1{
2 "truncate_dim": 768
3}| Metric | Value |
|---|---|
| cosine_accuracy@1 | 0.5307 |
| cosine_accuracy@3 | 0.7381 |
| cosine_accuracy@5 | 0.8145 |
| cosine_accuracy@10 | 0.884 |
| cosine_precision@1 | 0.5307 |
| cosine_precision@3 | 0.246 |
| cosine_precision@5 | 0.1629 |
| cosine_precision@10 | 0.0884 |
| cosine_recall@1 | 0.5307 |
| cosine_recall@3 | 0.7381 |
| cosine_recall@5 | 0.8145 |
| cosine_recall@10 | 0.884 |
| cosine_ndcg@10 | 0.7091 |
| cosine_mrr@10 | 0.6529 |
| cosine_map@100 | 0.6566 |
anchor and positive| anchor | positive | |
|---|---|---|
| type | string | string |
| details |
|
|
| anchor | positive |
|---|---|
Poly Studio X30 Halterung VESA Wandmontage | Poly Studio X30 VESA and Wall Mount. |
ALOGIC Elements Pro USB-C zu USB-A Kabel | Das ALOGIC Elements Pro USB-C zu USB-A Kabel ermöglicht es Ihnen, die neuesten USB-C Geräte wie Telefone, Tablets und Laptops mit Ihrem kompatiblen Zubehör oder Peripheriegerät zu verbinden. |
Equip VGA Splitter Signal-Bandbreite 450MHz | Der Video Splitter bietet eine Signal-Bandbreite von 450MHz. |
MatryoshkaLoss with these parameters:
1{
2 "loss": "MultipleNegativesRankingLoss",
3 "matryoshka_dims": [
4 1024,
5 768
6 ],
7 "matryoshka_weights": [
8 1,
9 1
10 ],
11 "n_dims_per_step": -1
12}eval_strategy: epochper_device_train_batch_size: 64per_device_eval_batch_size: 16gradient_accumulation_steps: 4learning_rate: 2e-05num_train_epochs: 4lr_scheduler_type: cosinewarmup_ratio: 0.1bf16: Truetf32: Trueload_best_model_at_end: Trueoptim: adamw_torch_fusedbatch_sampler: no_duplicatesoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: epochprediction_loss_only: Trueper_device_train_batch_size: 64per_device_eval_batch_size: 16per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 4eval_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: 4max_steps: -1lr_scheduler_type: cosinelr_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: Truefp16: Falsefp16_opt_level: O1half_precision_backend: autobf16_full_eval: Falsefp16_full_eval: Falsetf32: Truelocal_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: Trueignore_data_skip: Falsefsdp: []fsdp_min_num_params: 0fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}tp_size: 0fsdp_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_torch_fusedoptim_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: 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 | dim_1024_cosine_ndcg@10 | dim_768_cosine_ndcg@10 |
|---|---|---|---|---|
| 0.0388 | 1 | 2.9827 | - | - |
| 0.0777 | 2 | 3.3738 | - | - |
| 0.1165 | 3 | 3.7603 | - | - |
| 0.1553 | 4 | 3.826 | - | - |
| 0.1942 | 5 | 3.7338 | - | - |
| 0.2330 | 6 | 3.3327 | - | - |
| 0.2718 | 7 | 3.0444 | - | - |
| 0.3107 | 8 | 2.2803 | - | - |
| 0.3495 | 9 | 3.3083 | - | - |
| 0.3883 | 10 | 2.9835 | - | - |
| 0.4272 | 11 | 2.4352 | - | - |
| 0.4660 | 12 | 2.1565 | - | - |
| 0.5049 | 13 | 2.6124 | - | - |
| 0.5437 | 14 | 2.264 | - | - |
| 0.5825 | 15 | 1.9145 | - | - |
| 0.6214 | 16 | 1.8587 | - | - |
| 0.6602 | 17 | 1.4001 | - | - |
| 0.6990 | 18 | 1.8256 | - | - |
| 0.7379 | 19 | 1.1961 | - | - |
| 0.7767 | 20 | 1.3109 | - | - |
| 0.8155 | 21 | 1.5597 | - | - |
| 0.8544 | 22 | 1.4735 | - | - |
| 0.8932 | 23 | 1.0223 | - | - |
| 0.9320 | 24 | 1.1257 | - | - |
| 0.9709 | 25 | 1.3598 | - | - |
| 1.0 | 26 | 1.1203 | - | - |
| 1.0097 | 27 | 0.0 | 0.6978 | 0.6932 |
| 1.0388 | 28 | 0.7806 | - | - |
| 1.0777 | 29 | 1.3211 | - | - |
| 1.1165 | 30 | 1.4871 | - | - |
| 1.1553 | 31 | 0.935 | - | - |
| 1.1942 | 32 | 1.7934 | - | - |
| 1.2330 | 33 | 1.1227 | - | - |
| 1.2718 | 34 | 1.3105 | - | - |
| 1.3107 | 35 | 1.103 | - | - |
| 1.3495 | 36 | 1.3717 | - | - |
| 1.3883 | 37 | 0.9901 | - | - |
| 1.4272 | 38 | 1.3036 | - | - |
| 1.4660 | 39 | 1.2308 | - | - |
| 1.5049 | 40 | 1.2515 | - | - |
| 1.5437 | 41 | 1.1814 | - | - |
| 1.5825 | 42 | 1.2111 | - | - |
| 1.6214 | 43 | 0.9332 | - | - |
| 1.6602 | 44 | 1.3395 | - | - |
| 1.6990 | 45 | 0.7583 | - | - |
| 1.7379 | 46 | 1.3086 | - | - |
| 1.7767 | 47 | 0.9326 | - | - |
| 1.8155 | 48 | 0.9746 | - | - |
| 1.8544 | 49 | 0.6618 | - | - |
| 1.8932 | 50 | 0.7228 | - | - |
| 1.9320 | 51 | 0.7546 | - | - |
| 1.9709 | 52 | 1.0044 | - | - |
| 2.0 | 53 | 0.6009 | - | - |
| 2.0097 | 54 | 0.0467 | 0.7122 | 0.7100 |
| 2.0388 | 55 | 0.9867 | - | - |
| 2.0777 | 56 | 0.9411 | - | - |
| 2.1165 | 57 | 0.8141 | - | - |
| 2.1553 | 58 | 0.743 | - | - |
| 2.1942 | 59 | 1.0353 | - | - |
| 2.2330 | 60 | 1.2375 | - | - |
| 2.2718 | 61 | 0.9801 | - | - |
| 2.3107 | 62 | 1.2372 | - | - |
| 2.3495 | 63 | 0.8672 | - | - |
| 2.3883 | 64 | 1.0209 | - | - |
| 2.4272 | 65 | 0.8059 | - | - |
| 2.4660 | 66 | 0.8108 | - | - |
| 2.5049 | 67 | 1.1173 | - | - |
| 2.5437 | 68 | 1.2396 | - | - |
| 2.5825 | 69 | 0.7141 | - | - |
| 2.6214 | 70 | 0.9623 | - | - |
| 2.6602 | 71 | 0.7726 | - | - |
| 2.6990 | 72 | 1.0766 | - | - |
| 2.7379 | 73 | 0.8263 | - | - |
| 2.7767 | 74 | 0.8879 | - | - |
| 2.8155 | 75 | 1.5984 | - | - |
| 2.8544 | 76 | 1.0657 | - | - |
| 2.8932 | 77 | 1.1301 | - | - |
| 2.9320 | 78 | 0.8932 | - | - |
| 2.9709 | 79 | 1.0989 | - | - |
| 3.0 | 80 | 0.7175 | - | - |
| 3.0097 | 81 | 0.0 | 0.7123 | 0.7106 |
| 3.0388 | 82 | 0.9822 | - | - |
| 3.0777 | 83 | 0.9128 | - | - |
| 3.1165 | 84 | 0.8309 | - | - |
| 3.1553 | 85 | 0.8732 | - | - |
| 3.1942 | 86 | 1.004 | - | - |
| 3.2330 | 87 | 0.8509 | - | - |
| 3.2718 | 88 | 1.3577 | - | - |
| 3.3107 | 89 | 1.3243 | - | - |
| 3.3495 | 90 | 0.7953 | - | - |
| 3.3883 | 91 | 1.0733 | - | - |
| 3.4272 | 92 | 0.821 | - | - |
| 3.4660 | 93 | 1.1915 | - | - |
| 3.5049 | 94 | 1.1763 | - | - |
| 3.5437 | 95 | 0.9508 | - | - |
| 3.5825 | 96 | 0.6898 | - | - |
| 3.6214 | 97 | 0.7401 | - | - |
| 3.6602 | 98 | 1.1549 | - | - |
| 3.6990 | 99 | 1.1053 | - | - |
| 3.7379 | 100 | 0.7245 | 0.7126 | 0.7091 |
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{kusupati2024matryoshka,
2 title={Matryoshka Representation Learning},
3 author={Aditya Kusupati and Gantavya Bhatt and Aniket Rege and Matthew Wallingford and Aditya Sinha and Vivek Ramanujan and William Howard-Snyder and Kaifeng Chen and Sham Kakade and Prateek Jain and Ali Farhadi},
4 year={2024},
5 eprint={2205.13147},
6 archivePrefix={arXiv},
7 primaryClass={cs.LG}
8}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}