SentenceTransformer(
(0): Transformer({'transformer_task': 'feature-extraction', 'modality_config': {'text': {'method': 'forward', 'method_output_name': 'last_hidden_state'}}, 'module_output_name': 'token_embeddings', 'architecture': 'BertModel'})
(1): Pooling({'embedding_dimension': 384, 'pooling_mode': 'mean', 'include_prompt': True})
(2): Normalize({})
)pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("emirhanyac/e5-small-msmarco-tr")
5# Run inference
6queries = [
7 'query: Safra kesesi karaciğerin______ marjında bulunur',
8]
9documents = [
10 'passage: Karaciğer, karın boşluğunun üst kısmında diyaframın hemen altında bulunur. Karaciğerin büyük bir kısmı sağ kostal marjın altında bulunur ve ayrıca sol hemidiafragma ulaşmak için sola uzanır. Diyafram karaciğeri pleura, akciğerler, perikardiyum ve kalpten ayırır.',
11 "passage: Jay Gatsby (doğum adı James Gatz), F. Scott Fitzgerald'ın 1925 tarihli romanı The Great Gatsby'nin baş karakteridir. Gatsby, West Egg'deki Gotik bir konakta yaşayan inanılmaz derecede zengin bir genç adamdır.",
12 'passage: ağaç cerrahisi. n. Budama veya dalları destekleyerek hastalıklı veya hasarlı ağaçların tedavisi. ağaç cerrahı n.',
13]
14query_embeddings = model.encode_query(queries)
15document_embeddings = model.encode_document(documents)
16print(query_embeddings.shape, document_embeddings.shape)
17# [1, 384] [3, 384]
18
19# Get the similarity scores for the embeddings
20similarities = model.similarity(query_embeddings, document_embeddings)
21print(similarities)
22# tensor([[ 0.4641, -0.0035, 0.1333]])msmarco-tr-evalInformationRetrievalEvaluator| Metric | Value |
|---|---|
| cosine_accuracy@1 | 0.833 |
| cosine_accuracy@3 | 0.914 |
| cosine_accuracy@5 | 0.932 |
| cosine_accuracy@10 | 0.951 |
| cosine_precision@1 | 0.833 |
| cosine_precision@3 | 0.3047 |
| cosine_precision@5 | 0.1864 |
| cosine_precision@10 | 0.0951 |
| cosine_recall@1 | 0.833 |
| cosine_recall@3 | 0.914 |
| cosine_recall@5 | 0.932 |
| cosine_recall@10 | 0.951 |
| cosine_ndcg@10 | 0.8955 |
| cosine_mrr@10 | 0.8773 |
| cosine_map@100 | 0.8786 |
anchor and positive| anchor | positive | |
|---|---|---|
| type | string | string |
| modality | text | text |
| details |
|
|
| anchor | positive |
|---|---|
query: asal meridyen arktik daire veya antarktik daireden geçer mi | passage: Dünya'daki boylamın her meridyeni Antarktika Çemberi, Arktik Çemberi ve Dünya'daki diğer tüm enlem paralellerini geçer. Her ikisine de evet. Asal meridyen bir boylam çizgisidir ve bu nedenle hem Kuzey Kutbu hem de Antarktika çemberlerinden geçer, bunlar latent çizgileridir. |
query: Kaç test şeridi için ilaç ödeyecek | passage: Gary'ye göre, Medicare tarafından kapsanan test şeritlerinin sayısı, hastanın insülin kullanıp kullanmadığına bağlı olabilir. Bir yararlanıcı insülin kullanırsa, her ay 100 test şeridine ve lancet'a ve her 6 ayda bir lancet cihazına kadar çıkabilir, dedi Gary. |
query: Bir kedinin ortalama ömrü | passage: Köpeklerin ve Kedilerin Yaşam Beklentisi. Evcil hayvanlarımızın uzun yıllar boyunca hayatımızın bir parçası olacağını umuyoruz, ancak gerçek şu ki, ortalama yaşam beklentisi köpekler için 10-12 yıl ve kediler için 10-14 yıldır. Yaşlı evcil hayvanların çok özel ihtiyaçları vardır ve daha sonraki yıllarda kanser, artrit ve diş hastalıklarına özellikle duyarlıdırlar. |
MultipleNegativesRankingLoss with these parameters:
1{
2 "scale": 20.0,
3 "similarity_fct": "cos_sim",
4 "gather_across_devices": false,
5 "directions": [
6 "query_to_doc"
7 ],
8 "partition_mode": "joint",
9 "hardness_mode": null,
10 "hardness_strength": 0.0
11}per_device_train_batch_size: 64num_train_epochs: 1learning_rate: 2e-05warmup_steps: 0.1batch_sampler: no_duplicatesper_device_train_batch_size: 64num_train_epochs: 1max_steps: -1learning_rate: 2e-05lr_scheduler_type: linearlr_scheduler_kwargs: Nonewarmup_steps: 0.1optim: adamw_torch_fusedoptim_args: Noneweight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08optim_target_modules: Nonegradient_accumulation_steps: 1average_tokens_across_devices: Truemax_grad_norm: 1.0label_smoothing_factor: 0.0bf16: Falsefp16: Falsebf16_full_eval: Falsefp16_full_eval: Falsetf32: Nonegradient_checkpointing: Falsegradient_checkpointing_kwargs: Nonetorch_compile: Falsetorch_compile_backend: Nonetorch_compile_mode: Noneuse_liger_kernel: Falseliger_kernel_config: Noneuse_cache: Falseneftune_noise_alpha: Nonetorch_empty_cache_steps: Noneauto_find_batch_size: Falselog_on_each_node: Truelogging_nan_inf_filter: Trueinclude_num_input_tokens_seen: nolog_level: passivelog_level_replica: warningdisable_tqdm: Falseproject: huggingfacetrackio_space_id: Nonetrackio_bucket_id: Nonetrackio_static_space_id: Noneper_device_eval_batch_size: 8prediction_loss_only: Trueeval_on_start: Falseeval_do_concat_batches: Trueeval_use_gather_object: Falseeval_accumulation_steps: Noneinclude_for_metrics: []batch_eval_metrics: Falsesave_only_model: Falsesave_on_each_node: Falseenable_jit_checkpoint: Falsepush_to_hub: Falsehub_private_repo: Nonehub_model_id: Nonehub_strategy: every_savehub_always_push: Falsehub_revision: Noneload_best_model_at_end: Falseignore_data_skip: Falserestore_callback_states_from_checkpoint: Falsefull_determinism: Falseseed: 42data_seed: Noneuse_cpu: Falseaccelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}parallelism_config: Nonedataloader_drop_last: Falsedataloader_num_workers: 0dataloader_pin_memory: Truedataloader_persistent_workers: Falsedataloader_prefetch_factor: Noneremove_unused_columns: Truelabel_names: Nonetrain_sampling_strategy: randomlength_column_name: lengthddp_find_unused_parameters: Noneddp_bucket_cap_mb: Noneddp_broadcast_buffers: Falseddp_static_graph: Noneddp_backend: Noneddp_timeout: 1800fsdp: Nonefsdp_config: Nonedeepspeed: Nonedebug: []skip_memory_metrics: Truedo_predict: Falseresume_from_checkpoint: Nonewarmup_ratio: Nonelocal_rank: -1prompts: Nonebatch_sampler: no_duplicatesmulti_dataset_batch_sampler: proportionalrouter_mapping: {}learning_rate_mapping: {}| Epoch | Step | Training Loss | msmarco-tr-eval_cosine_ndcg@10 |
|---|---|---|---|
| -1 | -1 | - | 0.8841 |
| 0.1592 | 25 | 1.3794 | - |
| 0.3185 | 50 | 0.1740 | - |
| 0.4777 | 75 | 0.1527 | - |
| 0.6369 | 100 | 0.1369 | - |
| 0.7962 | 125 | 0.1406 | - |
| 0.9554 | 150 | 0.1435 | - |
| -1 | -1 | - | 0.8955 |
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{oord2019representationlearningcontrastivepredictive,
2 title={Representation Learning with Contrastive Predictive Coding},
3 author={Aaron van den Oord and Yazhe Li and Oriol Vinyals},
4 year={2019},
5 eprint={1807.03748},
6 archivePrefix={arXiv},
7 primaryClass={cs.LG},
8 url={https://arxiv.org/abs/1807.03748},
9}