Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks
Paper • 1908.10084 • Published • 13
How to use JonasGru/lt-e5 with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("JonasGru/lt-e5")
sentences = [
"query: eigulių kapinės paaiškinimas",
"passage: fair antigua we salute thee nuostabioji antigva mes sveikinam tave antigvos ir barbudos valstybinis himnas himno kompozitorius walter garnet picart chambers teksto autorius novelle hamilton richards naudojamas nuo oficialiai patvirtintas naudojamas kartu su karališkuoju himnu god save the queen dažniausiai grojamas tik pirmas himno posmas eilės anglų kalba fair antigua and barbuda we thy sons and daughters stand strong and firm in peace or danger to safeguard our native land we commit ourselves to building true nation brave and free ever striving ever seeking dwell in love and unity nuorodos instrumentinė versija mp kategorija antigvos ir barbudos atributika kategorija valstybiniai himnai",
"passage: eigulių kapinių vakarinė pusė eigulių kapinės kapinės kaune eiguliuose neries kairiajame krante jonavos netoli magistralės plotas ha kapinės suskirstytos kvartalus aptvertos metaline tvora sujungta mūro stulpais pagrindiniai vartai iš jonavos pusės antkapiniai paminklai daugiausia iš akmens kapinėse palaidota nemažai žymių žmonių įrengtas lietuvoje cirko artistų panteonas istorija eigulių kaimo kapinės įsteigtos apie iš pradžių laidoti kaimo ir apylinkės gyventojai eigulius ir prijungus prie kauno tapo miesto kapinėmis kapinėse yra lietuvos cirko artistų panteonas eigulių kapinės įrašytos kultūros paminklų sąrašą kapinėse palaidoti cirko artistai francas černiauskas pranas gudauskas maksimilianas truci massimiliano truzzi konstantinas feroni mykolas vilenčikas antanas pilkauskas jonas ramanauskas vladislava variakojienė stankutė jadvyga ramanauskienė stankutė ir kt lietuvos teatro kūrėjas juozas vaičkus aktorės bronė kurmytė monkevičienė ona kurmytė mazurkevičienė visuomenės kultūros ir sporto veikėjai juozas bagdonas stepas butautas klemensas čerbulėnas valerija čiurlionytė karužienė saulius gricius dalia grinkevičiūtė juozas muralis karininkai jonas jurgis bulota antanas malijonis anatolijus okulič kozarinas jonas petruitis rapolas okulič kazarinas pulkininkas archeologas petras tarasenka pedagogė pianistė kompozitorė elena laumenskienė profesoriai konradas aleksa vytautas vaičiūnas žemaitis kunigas erikas laiconas prozininkas vertėjas kazys jankauskas skulptorius tapytojas matas menčinskas sopranas violeta sagaitytė kankinė elena spirgevičiūtė perlaidota iš senųjų kapinių perlaidota prie šv antano paduviečio bažnyčios prozininkas mečys rakauskas politinis bei visuomenės veikėjas juozapas jokubauskas naikinant kauno centrines kapines šiose kapinėse perlaidoti savanoriai kauno miesto ir apskrities komendantūros eilinis pranas eimutis",
"passage: permas pradžia ma pabaiga ma epochos lopingijus gvadalupis kizuralis px dimetrodon grandis fosilinis skeletas permas ma milijonai metų permas paleozojaus eros periodas tarptautinis žymėjimas anglų geologas seras roderick murchison tyrinėdamas uralo kalnus išskyrė šią naują sistemą ir pavadino permu pagal permės miesto pavadinimą permo periodas pasibaigė permo triaso išmirimu tai kartu ir paleozojaus eros riba per didžiausią žemės istorijoje išmirimą išnyko daugiau kaip jūrų gyvybės ir sausumos gyvybės manoma kad tokį išmirimą paskatino nukritęs meteoritas arba didelis vulkaninis aktyvumas išsiveržę sibiro trapai šis įvykis pakeitė žemės klimatą todėl išnyko labai daug floros ir faunos rūšių lietuvoje ankstyvojo permo nuogulos aptinkamos tik siaurame ruože per perloją ir kalvariją link lenkijos ir karaliaučiaus srities pasienio sąlygos permo laikotarpiu kontinentai susijungė superkontinentą pangėją žemėje buvo du okeanai paleotetis mažas praktiškai vidinis okeanas ir pantalasas taip pat skaitykite geologinė laiko skalė šaltiniai kategorija paleozojus"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]This is a sentence-transformers model finetuned from intfloat/multilingual-e5-base. 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.
SentenceTransformer(
(0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: XLMRobertaModel
(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()
)
First install the Sentence Transformers library:
pip install -U sentence-transformers
Then you can load this model and run inference.
from sentence_transformers import SentenceTransformer
# Download from the 🤗 Hub
model = SentenceTransformer("sentence_transformers_model_id")
# Run inference
sentences = [
'query: stasys šimkus paaiškinimas',
'passage: stasys šimkus vasario motiškiuose seredžiaus valsčius rusijos imperija spalio kaune kompozitorius chorų organizatorius ir dirigentas muzikos publicistas folkloristas pedagogas ir muzikos visuomenininkas biografija vaizdas stasys šimkus with son algis jpeg thumb left px stasys šimkus su sūnumi algiu jav stasys šimkus su karine uniforma gimė mažažemių valstiečių šeimoje nuo muzikos mokėsi pas vietos vargonininką juozo naujalio vargonininkų ir dirigentų kursuose kaune dainavo mėgėjų choruose nuo metų vargonininkavo čiobiškyje krokialaukyje skirsnemunėje nuo dalyvavo lietuviškųjų vakarų veikloje organizavo chorus rengė koncertus užrašinėjo liaudies dainas mokėsi varšuvos muzikos instituto vargonų klasėje pas prof suržinskį vedė varšuvos lietuvių draugijos chorą stažavosi peterburgo vadovavo lietuvių chorams be studijų dėstė muziką šv kotrynos bažnyčios katalikų gimnazijoje kaip chormeisteris bendradarbiavo su česlovu sasnausku lietuvių draugijos nukentėjusiems dėl karo šelpti pasiųstas jav rinkti aukų organizavo lietuvių chorus rengė koncertus statė operetes jav miestuose leido žurnalą muzika grįžęs lietuvą dalyvavo lietuvių meno kūrėjų draugijos veikloje buvo jos valdybos pirmininkas studijavo kompoziciją leipcigo ir berlyno klaipėdoje įkūrė muzikos mokyklą joje dėstė ir buvo jos direktorius čia subūrė simfoninį orkestrą valstybės teatro dirigentas vytauto didžiojo universiteto mišraus choro vadovas ir švietimo ministerijos muzikos inspektorius šimkus daug prisidėjo prie pirmųjų lietuvos dainų švenčių rengimo buvo jų dirigentas nuo kauno profesorius palaidotas petrašiūnų kapinėse stasys šimkus buvo įdomi asmenybė didelio temperamento ūmus neišsenkamos energijos dažnai būdavo geros nuotaikos nekreipdavo dėmesio smulkmenas karštai griebdavosi',
'passage: alma adamkienė vasario šiauliuose gegužės vilniuje lietuvių filantropė visuomenės veikėja lr prezidento valdo adamkaus žmona buvusi pirmoji šalies dama vaizdas alma adamkiene and laura bush jpg thumb left px alma adamkienė su laura buš knygos viršelis biografija adamkienės tėvas stasys nutautas buvo prekybininkas motina ona soblytė nutautienė talkino vyrui versle augo telšiuose tarybinei kariuomenei veržiantis lietuvą nutautų šeima sunkvežimiu išvyko šventają kur apsigyveno pas žvejį nuo kurios juos laivu išplaukė link švedijos bet nepasiekus kranto laivą sulaikė vokiečių kariai ir išplukdino rygą kur juos dviems savaitėms pasodino rygos kalėjimą iš ten juos nuplukdino prekybiniu laivu gdanską tada traukinius ir vokietiją pabėgėlių stovyklą alma adamkienė vokietijoje baigė eichšteto lietuvių gimnaziją ten susipažino su valdu adamkumi kur jis mokėsi lietuvių kalbos vėliau studijavo erlangeno universiteto filologijos fakultete bavarijoje adamkienės šeima persikėlė jungtines amerikos valstijas kur jos šeima turėjo giminių ilinojuje pati gyveno rokforde atvykusi jungtines amerikos valstijas alma adamkienė dirbo plieno gamyklos laboratorijoje laborante vėliau draudimo kompanijoje ištekėjo už valdo adamkaus nuo net dvidešimt penkerius metus ji vadovavo iš jav lietuvių bendruomenės aktyvisto juozo bačiūno įsigytai tabor farmos vasarvietei kuri tapo svarbiu lietuvybės centru čia daug metų vykdavo liberaliosios pakraipos lietuvių išeivių organizacijos santara šviesa suvažiavimai dar rudenį vykstant prezidento rinkimų kampanijai alma adamkienė nusprendė kad ir kaip baigtųsi rinkimai ji lietuvoje pradės labdaringą veiklą buvo įkurtas almos adamkienės labdaros ir paramos',
]
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]
sentence_0 and sentence_1| sentence_0 | sentence_1 | |
|---|---|---|
| type | string | string |
| details |
|
|
| sentence_0 | sentence_1 |
|---|---|
query: fiziologija |
passage: vaizdas claude bernard and his pupils oil painting after léon augus wellcome jpg thumb upright paveiksle vaizduojamas klodas bernaras šiuolaikinės fiziologijos pradininkas su savo mokiniais fiziologija gamta ir λόγος lógos mokslas protas gr φυσιολογία mokslas tiriantis gyvųjų organizmų fizikines biochemines ir informacijos perdirbimo funkcijas tradiciškai fiziologijos mokslas skirstomas augalų fiziologiją ir gyvūnų fiziologiją tačiau fiziologijos principai yra tie patys visiems organizmams pvz žinios apie mielių ląsteles gali būti pritaikomos ir žmogaus ląstelėms normalioji fiziologija nagrinėja sveiko organizmo funkcijas jų reguliavimo mechanizmus norint suprasti gyvo organizmo funkcijas būtina išmanyti jo struktūrą sandarą todėl fiziologija glaudžiai siejasi su anatomija histologija biochemija biofizika fiziologija siejasi ir su kitais mokslais zoologija genetika matematika chemija fizika fiziologijos tyrimo objektas gyvas organizmas ar jo struktūrinės dalys organų sistemos organai audiniai ląstelės fiziologija padeda suprasti sveiko organizmo veiklą jo sugebėjimą prisitaikyti prie nuolat kintančios aplinkos nepakitus vidinės terpės sudėčiai fiziologija sudaro teorinį ir praktinį pagrindą nagrinėjant organizme vykstančius patologinius procesus sporto fiziologija tiria organizmo struktūrą bei funkcijų pokyčius veikiant fiziniams krūviams joje taikoma koncepcija kad veikiant fiziniams krūviams organizmas prie jų adaptuojasi sporto fiziologijos raida sporto fiziologija gana jaunas mokslas tik xix pabaigoje londone buvo išleistas pirmas fizinių pratimų vadovėlis kurio autorius la grange xx pradžioje šį mokslą toliau vystė chopkinas nustatęs ryšį tarp raumenų veiklos ir laktato kaupimosi dilas įkūrė laboratoriją nuovargio kilmei tirti po pirmojo |
query: gūšra apibrėžimas |
passage: gūšra dar žinoma kaip milvydo ežeras ežeras vakarų lietuvoje telšių rajone apie km šiaurės rytus nuo varnių km rytus nuo janapolės varnių regioniniame parke ežeras ovalios formos ilgis iš šiaurės pietus km plotis km altitudė giliausia vieta ežeras telkšo ištirpusio ledo luisto suformuotame duburyje krantai žemi vakarinis krantas plačiai užpelkėjęs apaugęs mišku pietuose įteka bevardis upelis vakaruose netoliese tekančią virvytę išteka druja ežeras yra drujos draustinyje prie gūšros rasta neolito gyvenviečių pėdsakų šalia įsikūrę milvydiškių ir lenkalių kaimai šaltiniai kategorija telšių rajono ežerai |
query: get apibrėžimas |
passage: getas dgs gittim gittin žydų skyrybų raštas kurį žydas išduoda žydei nutraukiant vedybas pagrindinis rašto tekstas nuo dabar tu gali bendrauti su visais vyrais kitaip sakant tu nebesi ištekėjusi moteris ir tau paleistuvavimą draudžiantys įstatymai daugiau nebegalioja geto skyrybų raštas taip pat sugrąžina moteriai visas buvusias teises kurias ji turėjo prieš ištekėdama būtini reikalavimai getui getas skyrybų raštas privalo būti parašytas tik pas judaizmo soferį tam turi būti gautas judėjo vyro sutikimas žydei žmonai draudžiama skirtis getas turi būti parašytas ant tokio poperiaus kurio turinio nebūtų galima ištrinti arba taisyti geto data turi atitikti skyrybų rašto sudarymo datą bet koks geto taisyklių pažeidimas panaikina skyrybų procedūrą nuorodos skyrybos judėjiškai skyrybų raštas kategorija judaizmas |
MultipleNegativesRankingLoss with these parameters:{
"scale": 20.0,
"similarity_fct": "cos_sim"
}
per_device_train_batch_size: 12per_device_eval_batch_size: 12num_train_epochs: 1multi_dataset_batch_sampler: round_robinoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: noprediction_loss_only: Trueper_device_train_batch_size: 12per_device_eval_batch_size: 12per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 1eval_accumulation_steps: Nonelearning_rate: 5e-05weight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08max_grad_norm: 1num_train_epochs: 1max_steps: -1lr_scheduler_type: linearlr_scheduler_kwargs: {}warmup_ratio: 0.0warmup_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: Falsefp16_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: Falsehub_always_push: Falsegradient_checkpointing: Falsegradient_checkpointing_kwargs: Noneinclude_inputs_for_metrics: Falseeval_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: Falsebatch_sampler: batch_samplermulti_dataset_batch_sampler: round_robin| Epoch | Step | Training Loss |
|---|---|---|
| 0.0300 | 500 | 0.3228 |
| 0.0600 | 1000 | 0.0094 |
| 0.0900 | 1500 | 0.009 |
| 0.1200 | 2000 | 0.0037 |
| 0.1500 | 2500 | 0.005 |
| 0.1800 | 3000 | 0.0041 |
| 0.2100 | 3500 | 0.0034 |
| 0.2400 | 4000 | 0.0052 |
| 0.2700 | 4500 | 0.0036 |
| 0.3000 | 5000 | 0.0015 |
| 0.3300 | 5500 | 0.0009 |
| 0.3600 | 6000 | 0.0017 |
| 0.3900 | 6500 | 0.0023 |
| 0.4200 | 7000 | 0.002 |
| 0.4500 | 7500 | 0.0023 |
| 0.4800 | 8000 | 0.0017 |
| 0.5100 | 8500 | 0.0018 |
| 0.5400 | 9000 | 0.0018 |
| 0.5700 | 9500 | 0.0023 |
| 0.6000 | 10000 | 0.0019 |
| 0.6300 | 10500 | 0.0016 |
| 0.6600 | 11000 | 0.0007 |
| 0.6900 | 11500 | 0.0005 |
| 0.7200 | 12000 | 0.0012 |
| 0.7500 | 12500 | 0.0003 |
| 0.7800 | 13000 | 0.0007 |
| 0.8100 | 13500 | 0.0013 |
| 0.8400 | 14000 | 0.0007 |
| 0.8700 | 14500 | 0.0004 |
| 0.9000 | 15000 | 0.0002 |
| 0.9300 | 15500 | 0.0012 |
| 0.9600 | 16000 | 0.0004 |
| 0.9900 | 16500 | 0.0006 |
@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",
}
@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}
}
Base model
intfloat/multilingual-e5-base
from sentence_transformers import SentenceTransformer model = SentenceTransformer("JonasGru/lt-e5") sentences = [ "query: eigulių kapinės paaiškinimas", "passage: fair antigua we salute thee nuostabioji antigva mes sveikinam tave antigvos ir barbudos valstybinis himnas himno kompozitorius walter garnet picart chambers teksto autorius novelle hamilton richards naudojamas nuo oficialiai patvirtintas naudojamas kartu su karališkuoju himnu god save the queen dažniausiai grojamas tik pirmas himno posmas eilės anglų kalba fair antigua and barbuda we thy sons and daughters stand strong and firm in peace or danger to safeguard our native land we commit ourselves to building true nation brave and free ever striving ever seeking dwell in love and unity nuorodos instrumentinė versija mp kategorija antigvos ir barbudos atributika kategorija valstybiniai himnai", "passage: eigulių kapinių vakarinė pusė eigulių kapinės kapinės kaune eiguliuose neries kairiajame krante jonavos netoli magistralės plotas ha kapinės suskirstytos kvartalus aptvertos metaline tvora sujungta mūro stulpais pagrindiniai vartai iš jonavos pusės antkapiniai paminklai daugiausia iš akmens kapinėse palaidota nemažai žymių žmonių įrengtas lietuvoje cirko artistų panteonas istorija eigulių kaimo kapinės įsteigtos apie iš pradžių laidoti kaimo ir apylinkės gyventojai eigulius ir prijungus prie kauno tapo miesto kapinėmis kapinėse yra lietuvos cirko artistų panteonas eigulių kapinės įrašytos kultūros paminklų sąrašą kapinėse palaidoti cirko artistai francas černiauskas pranas gudauskas maksimilianas truci massimiliano truzzi konstantinas feroni mykolas vilenčikas antanas pilkauskas jonas ramanauskas vladislava variakojienė stankutė jadvyga ramanauskienė stankutė ir kt lietuvos teatro kūrėjas juozas vaičkus aktorės bronė kurmytė monkevičienė ona kurmytė mazurkevičienė visuomenės kultūros ir sporto veikėjai juozas bagdonas stepas butautas klemensas čerbulėnas valerija čiurlionytė karužienė saulius gricius dalia grinkevičiūtė juozas muralis karininkai jonas jurgis bulota antanas malijonis anatolijus okulič kozarinas jonas petruitis rapolas okulič kazarinas pulkininkas archeologas petras tarasenka pedagogė pianistė kompozitorė elena laumenskienė profesoriai konradas aleksa vytautas vaičiūnas žemaitis kunigas erikas laiconas prozininkas vertėjas kazys jankauskas skulptorius tapytojas matas menčinskas sopranas violeta sagaitytė kankinė elena spirgevičiūtė perlaidota iš senųjų kapinių perlaidota prie šv antano paduviečio bažnyčios prozininkas mečys rakauskas politinis bei visuomenės veikėjas juozapas jokubauskas naikinant kauno centrines kapines šiose kapinėse perlaidoti savanoriai kauno miesto ir apskrities komendantūros eilinis pranas eimutis", "passage: permas pradžia ma pabaiga ma epochos lopingijus gvadalupis kizuralis px dimetrodon grandis fosilinis skeletas permas ma milijonai metų permas paleozojaus eros periodas tarptautinis žymėjimas anglų geologas seras roderick murchison tyrinėdamas uralo kalnus išskyrė šią naują sistemą ir pavadino permu pagal permės miesto pavadinimą permo periodas pasibaigė permo triaso išmirimu tai kartu ir paleozojaus eros riba per didžiausią žemės istorijoje išmirimą išnyko daugiau kaip jūrų gyvybės ir sausumos gyvybės manoma kad tokį išmirimą paskatino nukritęs meteoritas arba didelis vulkaninis aktyvumas išsiveržę sibiro trapai šis įvykis pakeitė žemės klimatą todėl išnyko labai daug floros ir faunos rūšių lietuvoje ankstyvojo permo nuogulos aptinkamos tik siaurame ruože per perloją ir kalvariją link lenkijos ir karaliaučiaus srities pasienio sąlygos permo laikotarpiu kontinentai susijungė superkontinentą pangėją žemėje buvo du okeanai paleotetis mažas praktiškai vidinis okeanas ir pantalasas taip pat skaitykite geologinė laiko skalė šaltiniai kategorija paleozojus" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]