Views
No views yet
SentenceTransformer(
(0): Transformer({'max_seq_length': 128, 'do_lower_case': False}) with Transformer model: BertModel
(1): Pooling({'word_embedding_dimension': 384, '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()
)pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("FareedKhan/flax-sentence-embeddings_all_datasets_v4_MiniLM-L6_FareedKhan_prime_synthetic_data_2k_10_32")
5# Run inference
6sentences = [
7 '\nAtypical hemolytic uremic syndrome (aHUS) with H factor anomaly is a disease characterized by an atypical form of hemolytic uremic syndrome, a severe thrombotic microangiopathy that leads to kidney failure, anemia, and thrombocytopenia. This specific subtype of aHUS is notable for its association with an anomaly in the H factor, potentially involving complement system dysregulation. As such, it falls under the broader category of hemolytic uremic syndrome, a condition marked by differential diagnosis complexity and distinct etiologies. Patients with aHUS often require a nuanced approach to diagnosis and management, emphasizing awareness of its distinct characteristics in comparison with other forms of hemolytic uremic syndrome, ensuring comprehensive and accurate differential diagnosis which might include conditions like thrombotic thrombocytopenic purpura (TTP) or disseminated intravascular coagulation (DIC). The identification and management of aHUS with H factor anomaly necessitates multidisciplinary collaboration and up-to-date knowledge alongside genetic and clinical features specific to this condition.',
8 'Could you list the diseases related to or subtypes of type 1 atypical hemolytic uremic syndrome for differential diagnosis purposes?',
9 'Which diseases are associated with anomalies in the CD4 gene or protein, alongside genetic mutations that impact muscle protein synthesis?',
10]
11embeddings = model.encode(sentences)
12print(embeddings.shape)
13# [3, 384]
14
15# Get the similarity scores for the embeddings
16similarities = model.similarity(embeddings, embeddings)
17print(similarities.shape)
18# [3, 3]dim_384InformationRetrievalEvaluator| Metric | Value |
|---|---|
| cosine_accuracy@1 | 0.3861 |
| cosine_accuracy@3 | 0.4604 |
| cosine_accuracy@5 | 0.4901 |
| cosine_accuracy@10 | 0.5149 |
| cosine_precision@1 | 0.3861 |
| cosine_precision@3 | 0.1535 |
| cosine_precision@5 | 0.098 |
| cosine_precision@10 | 0.0515 |
| cosine_recall@1 | 0.3861 |
| cosine_recall@3 | 0.4604 |
| cosine_recall@5 | 0.4901 |
| cosine_recall@10 | 0.5149 |
| cosine_ndcg@10 | 0.4514 |
| cosine_mrr@10 | 0.4312 |
| cosine_map@100 | 0.4383 |
positive and anchor| positive | anchor | |
|---|---|---|
| type | string | string |
| details |
|
|
| positive | anchor |
|---|---|
[object Object][object Object]Epilepsy is a neurological disorder characterized by recurrent seizures, which can be sudden, abnormal electrical events in the brain. Seizures can affect different parts of the brain and range from mild to severe. Symptoms can include muscle stiffness, twitching, loss of consciousness, and cognitive disruptions. Seizures can be divided into focal (partial) seizures and generalized seizures.[object Object][object Object]### Causes:[object Object]1. [object Object]: These can lead to scar tissue and abnormal electrical activity.[object Object]2. [object Object]: Genetic or developmental issues can cause seizures.[object Object]3. [object Object]: These can result in seizures.[object Object]4. [object Object]: Some genetic conditions lead to epilepsy.[object Object][object Object]### Complications:[object Object]- [object Object]: Continuous seizure activity lasting more than five minutes.[object Object]- [object Object]: Unexplained death during an untreated condition, especially if seizures aren't controlled.[object Object]- [object Object]: Increased risk for depression, anxiety, and | Search for medical conditions not treatable by any known medications that present with hoarseness as a symptom. |
[object Object]Diphyllobothriasis, also known as bothriocephalosis, is a parasitosis caused by the intestinal infection with the larval stage of the tapeworm Diphyllobothrium. This condition is characterized by a broad array of symptoms, including frequent stomach discomfort, nausea, appetite loss, fatigue, and weakness. These symptoms are medically attributed to anemia, which stems from vitamin B12 deficiency—a common complication linked to this parasitosis. The anemia caused by diphyllobothriasis can also resemble Biermer's anemia, distinguished by abnormally large red blood cells. Individuals with a family history of ceestode infections, such as diphyllobothriasis, and those who exhibit symptoms such as those described, may be more susceptible to this condition. The disease, which is cosmopolitan in nature, has been reported in Europe, primarily in areas like the Italian, Swiss, and French Alps, though its prevalence across the continent remains unknown. Treatment for diphyllobothriasis typically involves the use of standard medications such as niclosamide or praziquantel, which are effective in clearing the parasite. | What could be the condition causing frequent stomach discomfort, nausea, appetite loss, fatigue, and weakness in me, possibly linked to a family history of Cestode infection and associated with vitamin B12 deficiency and abnormal red blood cells resembling Biermer's anemia symptoms? |
[object Object]The provided list appears to be a collection of gene names. Genes are segments of DNA that code for proteins and play a crucial role in various biological functions, influencing traits, growth, and processes within an organism. They are fundamental units of heredity. The presence of these gene names suggests that the document is most likely related to genetic research, medical studies, or bioinformatics. This could involve analyses of genetic sequences, expression patterns, or functional assays related to the specific genes mentioned, possibly with the aim of understanding genetic disorders, development, or disease mechanisms. | Which cellular structures engage in interactions with genes or proteins that are affected by the administration of Mevastatin? |
MatryoshkaLoss with these parameters:
1{
2 "loss": "MultipleNegativesRankingLoss",
3 "matryoshka_dims": [
4 384
5 ],
6 "matryoshka_weights": [
7 1
8 ],
9 "n_dims_per_step": -1
10}eval_strategy: epochper_device_train_batch_size: 32learning_rate: 1e-05num_train_epochs: 10warmup_ratio: 0.1bf16: Truetf32: Falseload_best_model_at_end: Trueoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: epochprediction_loss_only: Trueper_device_train_batch_size: 32per_device_eval_batch_size: 8per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 1eval_accumulation_steps: Nonetorch_empty_cache_steps: Nonelearning_rate: 1e-05weight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08max_grad_norm: 1.0num_train_epochs: 10max_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: Truefp16: Falsefp16_opt_level: O1half_precision_backend: autobf16_full_eval: Falsefp16_full_eval: Falsetf32: Falselocal_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}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: Falseeval_on_start: Falseuse_liger_kernel: Falseeval_use_gather_object: Falsebatch_sampler: batch_samplermulti_dataset_batch_sampler: proportional| Epoch | Step | Training Loss | dim_384_cosine_map@100 |
|---|---|---|---|
| 0 | 0 | - | 0.3748 |
| 0.1754 | 10 | 1.5606 | - |
| 0.3509 | 20 | 1.5914 | - |
| 0.5263 | 30 | 1.6623 | - |
| 0.7018 | 40 | 1.7258 | - |
| 0.8772 | 50 | 1.6031 | - |
| 1.0 | 57 | - | 0.4241 |
| 1.0526 | 60 | 1.4494 | - |
| 1.2281 | 70 | 1.4091 | - |
| 1.4035 | 80 | 1.3177 | - |
| 1.5789 | 90 | 1.3299 | - |
| 1.7544 | 100 | 1.459 | - |
| 1.9298 | 110 | 1.3534 | - |
| 2.0 | 114 | - | 0.4214 |
| 2.1053 | 120 | 1.3023 | - |
| 2.2807 | 130 | 1.2222 | - |
| 2.4561 | 140 | 1.2191 | - |
| 2.6316 | 150 | 1.0443 | - |
| 2.8070 | 160 | 1.1894 | - |
| 2.9825 | 170 | 1.0955 | - |
| 3.0 | 171 | - | 0.4156 |
| 3.1579 | 180 | 1.1698 | - |
| 3.3333 | 190 | 0.9699 | - |
| 3.5088 | 200 | 1.0524 | - |
| 3.6842 | 210 | 0.9902 | - |
| 3.8596 | 220 | 1.0943 | - |
| 4.0 | 228 | - | 0.4221 |
| 4.0351 | 230 | 0.9793 | - |
| 4.2105 | 240 | 0.9786 | - |
| 4.3860 | 250 | 1.0352 | - |
| 4.5614 | 260 | 0.9809 | - |
| 4.7368 | 270 | 0.8568 | - |
| 4.9123 | 280 | 0.9372 | - |
| 5.0 | 285 | - | 0.4264 |
| 5.0877 | 290 | 0.8529 | - |
| 5.2632 | 300 | 0.9472 | - |
| 5.4386 | 310 | 0.8436 | - |
| 5.6140 | 320 | 0.8166 | - |
| 5.7895 | 330 | 0.8731 | - |
| 5.9649 | 340 | 0.9489 | - |
| 6.0 | 342 | - | 0.4274 |
| 6.1404 | 350 | 0.9991 | - |
| 6.3158 | 360 | 0.7533 | - |
| 6.4912 | 370 | 0.9122 | - |
| 6.6667 | 380 | 0.8404 | - |
| 6.8421 | 390 | 0.7928 | - |
| 7.0 | 399 | - | 0.4302 |
| 7.0175 | 400 | 0.8332 | - |
| 7.1930 | 410 | 0.7534 | - |
| 7.3684 | 420 | 0.8424 | - |
| 7.5439 | 430 | 0.8465 | - |
| 7.7193 | 440 | 0.8461 | - |
| 7.8947 | 450 | 0.7203 | - |
| 8.0 | 456 | - | 0.4344 |
| 8.0702 | 460 | 0.8144 | - |
| 8.2456 | 470 | 0.7895 | - |
| 8.4211 | 480 | 0.7665 | - |
| 8.5965 | 490 | 0.883 | - |
| 8.7719 | 500 | 0.6908 | - |
| 8.9474 | 510 | 0.8481 | - |
| 9.0 | 513 | - | 0.4365 |
| 9.1228 | 520 | 0.7521 | - |
| 9.2982 | 530 | 0.6971 | - |
| 9.4737 | 540 | 0.7081 | - |
| 9.6491 | 550 | 0.8272 | - |
| 9.8246 | 560 | 0.7922 | - |
| 10.0 | 570 | 0.7998 | 0.4383 |
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}