mpnet_jacket4k / README.md
knguyennguyen's picture
Add new SentenceTransformer model.
18a57a7 verified
metadata
tags:
  - sentence-transformers
  - sentence-similarity
  - feature-extraction
  - generated_from_trainer
  - dataset_size:7598
  - loss:MultipleNegativesRankingLoss
base_model: sentence-transformers/all-mpnet-base-v2
widget:
  - source_sentence: >-
      men's jacket featuring a waterproof outer layer, an insulated inner layer,
      and multiple secure pockets for storage.
    sentences:
      - >-
        Title: Pure Leather Biker Short Jacket Black Men’s Bike Jackets -
        Cowhide Top Grain Classic Belted Strap Descripion: ['We Believe in
        Beauty of Simplicity' 'PURE LEATHER'
         'jackets are negligible and made for regular wear. Our plans are ageless any style without overwhelming it, in light of the fact that there is'
         'Beauty in Simplicity' '.' 'Product Detail:'
         'This black leather jacket for men is traditional yet contemporary and useful ranging from your weekend sport coat. With great style, this cowhide leather cool leather jackets for men will keep you warm and comfortable in all types of weather. The style is similar to our PR-40 in Natural Sheep Leather but this one comes in a different leather altogether Cow Natural. Molds need to be removed because they are prone to breeding when three Parts of it are in excess. Parts are dirt, its multiplicity, and temperature. Firstly, you have to hang it on a hanger in a nicely-ventilated place.'
         'Note:'
         'The color will become slightly darker due to its transparent appearance. Do not stretch it for extended time as it will fall prey to different proportions. Jackets from Pure Leather are made by caring people who love their jobs, so you won’t get annoyed or disappointed when wearing them!! (Do Not Wash. Do Not Bleach. Do Not Tumble. Do Not Iron, Dry Clean only- Satisfaction guaranteed'
         'A Great Gift:'
         'These beautiful Jackets Would be a perfect gift for that special someone in your life. Buy these jackets for your Husband, Son, Brother, or best friend, and without a doubt you have purchased the perfect present for the any occasion, whether it is for Father’s Day, Valentine-day, Christmas, Graduation or their Birthday.'
         'Material:' 'Cowhide Leather' 'Colors:' 'Black' 'Click' 'ADD TO CART'
         'to order your Ancient Jacket' 'TODAY!']
      - >-
        Title: Pendleton Men's Jacquard Sherpa-Lined Shirt Jacket Descripion:
        ['New style for Pendleton but using our iconic Harding print. Taking our
        classic shirt jacket and lining it with soft sherpa for warmth.']
      - >-
        Title: THE NORTH FACE Men's Clement Triclimate Jacket Descripion: ['The
        North Face Size Chart' 'The North Face Size Chart'
         'Please note, the logo and hardware color may vary in styles marked as Prior Season.'
         'Please note, the logo and hardware color may vary in styles marked as Prior Season.'
         'Your go-to jacket all season long, this versatile The North Face® Clement Triclimate Jacket system pairs a breathable, waterproof shell with a warm, insulated liner jacket'
         'Your go-to jacket all season long, this versatile The North Face® Clement Triclimate Jacket system pairs a breathable, waterproof shell with a warm, insulated liner jacket'
         'Waterproof, insulated three-in-one ski jacket.'
         'Waterproof, insulated three-in-one ski jacket.'
         'Removable, helmet-compatible hood.' 'Removable, helmet-compatible hood.'
         'Secure-zip chest and hand pockets.' 'Secure-zip chest and hand pockets.'
         'Internal goggle pocket.' 'Internal goggle pocket.'
         'Underarm vents for added breathability.'
         'Underarm vents for added breathability.'
         'Secure-zip wrist pocket with goggle wipe.'
         'Secure-zip wrist pocket with goggle wipe.' 'Zip-in integration.'
         'Zip-in integration.' 'Secure-zip hand pockets.'
         'Secure-zip hand pockets.' '100% polyester.' '100% polyester.'
         '50D 73 G/M² 100% Recycled Polyester With Non-PFC DWR Finish.'
         '50D 73 G/M² 100% Recycled Polyester With Non-PFC DWR Finish.'
         'Machine wash cold, hang dry.' 'Machine wash cold, hang dry.' 'Imported.'
         'Imported.']
  - source_sentence: a lightweight performance jacket for outdoor activities
    sentences:
      - >-
        Title: Puma Golf Men's First Mile Wind Jacket Descripion: ['Head turner.
        Impact maker. The First Mile Wind Jacket is all about the important
        details—from its classic style to sustainable materials. The jacket uses
        yarn from recycled plastic bottles, sustainably sourced from communities
        of collectors.']
      - >-
        Title: Tommy Hilfiger Men's Lightweight Performance Softshell Hoody
        Jacket Descripion: ['Lightweight perfomance jacket in a breathable and
        water resistant softshell with microfleece backing center front zipper
        closure adjustable drawstring hood adjustable snap cuffs Tommy Hilfiger
        flag screenprint on left chest round silicon logo patch on left sleeve
        and two lower welt zipper pockets']
      - >-
        Title: Columbia Women's Pike Lake Long Jacket Descripion: ["Don't let
        cold weather ice out your plans. This puffer is ideal when the temp
        drops, combining a Storm-Lite shell for durability, a heat-holding
        thermal-reflective lining, and Thermarator insulation for warmth without
        bulk. Snaps at the side seams make it easy to move. Your new puffy
        favorite, complete with zippered hand pockets, interior security pocket,
        drawcord adjustable hood and hem, plus handy snap-adjustable side seams.
        This is the perfect long coat to manage the harshest of cold weather
        adventures. This women’s winter jacket is offered in multiple sizes and
        colors. Available in extended sizing. Relaxed and Regular Fit. To ensure
        the size you choose is right, utilize our sizing chart and the following
        measurement instructions: For the sleeves, start at the center back of
        your neck and measure across the shoulder and down to the sleeve. If you
        come up with a partial number, round up to the next even number. For the
        chest, measure at the fullest part of the chest, under the armpits and
        over the shoulder blades, keeping the tape measure firm and level."]
  - source_sentence: >-
      baby's outerwear jumpsuit with a hood, soft fabric, and a front zipper for
      easy access.
    sentences:
      - >-
        Title: Lykmera Baby Coat Toddler Kimono Solid Silk Robes Kids Clothes
        Sleepwear Bathrobe Girls Baby Satin Girls Coat Jacket Descripion:
        ["2.Casual style top, , cute and comfy baby clothes 3.Great idea for a
        baby clothes, there's no doubt in our mind your little baby will be the
        cutest Package include:1PC Bathrobe+1PC Ribbons 1.It is made of high
        quality materials,Soft hand feeling, no any harm to your baby skin
        Clothing Length:Regular Pattern Type:Solid Gender:Girls Please note that
        slight color difference should be acceptable due to the light and
        screen. Both hand wash and machine wash is OK Occasion:Casual
        Material:Polyester Attention plz: If your kid is , we recomend choosing
        a larger size, thanks."]
      - >-
        Title: Loloda Baby Girls Toddler Fur Faux Coat Vest Winter Warm Solid
        Color Sleeveless Waistcoat Jacket Descripion: ["Set Include: 1Pc Vest
        Condition: Brand New Material: Polyester Tag No.---|---Recommended
        Size---|------Chest------|------Length--- ---80-----|------9-12
        Months-----|---23.6''/60cm---|----13.8''/35cm ---90-----|-----12-18
        Months-----|---25.2''/64cm---|----14.2''/36cm --100-----|-----18-24
        Months-----|---26.0''/66cm---|----14.6''/37cm --110-----|----------2-3
        Years---------|---26.8''/68cm---|----15.0''/38cm
        --120-----|----------3-4
        Years---------|---27.6''/70cm---|----16.5''/42cm
        --130-----|----------5-6 Years---------|---28.3''/72cm---|---17.3''/44
        cm --140-----|----------7-8
        Years---------|---30.7''/78cm---|---19.3''/49 cm
        --150-----|---------9-10 Years---------|---31.5''/80cm---|---19.7''/50
        cm"]
      - >-
        Title: Baby's hooded jumpsuit outerwear toddler jacket, Cosplay zipper
        full-body jumpsuit Descripion: ['Product Description: Features: Cute
        animal hooded onesie, warm onesie tunic. High-grade facecloth fabric,
        soft to the touch, so that the baby is comfortable and cozy. Hooded
        design makes the baby warm all over. Anti-pinch collar design takes care
        of baby\'s neck. Suitable for loungewear, parties, holidays, festivals,
        photo shoots, etc. One-piece only, does not include any other
        accessories. Specification.: Color: fox (orange) / tiger (yellow) /
        rabbit (pink) optional Size: 80CM / 90CM / 100CM (optional) Note: 1. The
        size is just for your reference. Please check the measurements to choose
        the right size for your baby! Meanwhile, please choose the larger size
        because babies at the same age may have different height. 2. Please
        allow 1-3cm (0.4-1.18") difference due to manual measurement and slight
        color variation for different display setting Package Including: 1 x
        Jumpsuit']
  - source_sentence: a packable rain jacket for bowhunters during early season downpours
    sentences:
      - >-
        Title: OAKI Rain & Trail Suit - Adult One Piece Rain Jacket & Pant
        Orange, Medium Descripion: ['Oaki Adults One-Piece Waterproof Trail Rain
        Suit are tough and weather-resistant, giving your overall protection
        with ripstop fabric. This one-piece rain suit is made of Nylon Taslon PU
        coated fabric. Unlike rubber jumpsuits, this rain suit is breathable.
        Whether you’re camping or playing with your kids in the backyard, these
        comfortably breathable rainsuits will keep you dry.']
      - >-
        Title: HIDDEN MAKERS Womens Trendy Thickened Parka Jacket with Muskrat
        Fur & Belt Descripion: ['You can never be ready to face the winter
        chills unless you have your down puffed winter jacket, here’s one with a
        trendy twist to it!  A sight to behold, designed with an edgy twist to
        parka jackets, this one is surely gonna keep you warm and comfortable
        because of double front layering. The muskrat fur and detachable belt
        adds to the sleek silhouette of the long winter jacket. Padded with
        goose feather down to lock the heat, the slim-fit cut will add glamor to
        your walks down the frosty city streets.   COLOR VARIATIONS  Black  
        SIZE (inch)  · S : Shoulder 15.5, Chest 38.5, Waist 35.8, Bottom Width
        43.3, Body Length(back) 32.6, Sleeve Width 13.7, Sleeve Length 23.4  ·
        M<: Shoulder 15.9, Chest 41.3, Waist 38.1, Bottom Width 47.2, Body
        Length(back) 33.6, Sleeve Width 14.5, Sleeve Length 23.6  · L : Shoulder
        16.1, Chest 41.4, Waist 39.3 , Bottom Width 48, Body Length(back) 34,
        Sleeve Width 15.3, Sleeve Length 23.8   MODEL SIZE  · Height 5.6ft  
        MATERIAL & COMPOSITION  · Outshell & Lining : 100% polyester  ·
        Coloration : 100% muskrat fur  · Filling1 : 80% goose down & 20% goose
        feather  · Filling2 : 100% polyester   WASHING & KEEPING TIPS  · Wash
        inside out and dry clean or hand wash  · Do not use chlorine or oxygen
        detergents  · Avoid direct sunlight and dry In shade  · Wash dark colors
        separately  · Keep away from fire   DESIGN  · Snap button closure on
        front  · Two outside pockets  · Dual Layered front design to block wind 
        · 100% muskrat fur from hoodie to hem  · Removable fur on the hood  ·
        Detachable waist belt  · Windbreaker and waterproof fabric  · Suitable
        length and fit to allow optimum movement   MULTI PURPOSE  · Made of
        waterproof and windproof fabrics for everyday life  · It can be worn
        comfortably and warmly in everyday life']
      - >-
        Title: ASIO Gear Packable Rain Jacket Descripion: ['Ideal for keeping
        bowhunters dry during those early season pop-up downpours, this durable
        camo hunting jacket features pit zips for superior breathability, slips
        easily over ASIO Gear’s mid and late-season gear for ultimate rain
        protection, and, when paired with our packable hunting rain pants, packs
        down to the size of a softball. No matter what part of the whitetail
        range you hunt, you can be sure you’ll be super dry at the end of your
        sit with this packable hunting jacket (and when paired with our equally
        as awesome packable hunting rain pants).']
  - source_sentence: a pair of statement stud earrings for brightening up outfits
    sentences:
      - >-
        Title: Kendra Scott Sienna Ear Jacket Earrings Descripion: ['The Kendra
        Scott Sienna Ear Jacket Earrings are the best bright-colored statement
        stud. Straight post backing. Removeable jackets. Gold plated brass.
        Shell detailing. Remove jewelry when swimming, bathing, or exercising.
        Imported.']
      - >-
        Title: TrailCrest Infant-Toddler Boys & Girls Fleece Full Zip Mock Neck
        Soft Jacket Descripion: ['Feel the chill! On chilly summer nights out in
        the wilderness or crisp spring and autumn days, these semi-fitted fleece
        zips are the perfect cover to keep you warm. These fleece zips are great
        for camping, hiking and just about any sport related activities.
        Featuring bright accent colors with realistic camo print to make the
        experience all the more exciting. Fully equipped with a mock neck collar
        for extra warmth, two hand pockets with hidden zippers, an adjustable
        drawstring at the bottom, and elastic cuffs. All set for the outdoors!']
      - >-
        Title: Tainilawo Womens Hooies Long Sleeve Shirts Sweatershirt Giraffe
        Prints Hooded Pullover Overszied Coats Jackets Descripion: ["Size: M US:
        10 EU: 40 Bust: 100cm/39.37'' Shoulder: 47.5cm/18.70'' Sleeve:
        55.5cm/21.85'' Length: 61cm/24.02'' Size: L US: 12 EU: 42 Bust:
        104cm/40.94'' Shoulder: 49.5cm/19.49'' Sleeve: 57.5cm/22.64'' Length:
        63cm/24.80'' Size: XL US: 14 EU: 44 Bust: 108cm/42.52'' Shoulder:
        51.5cm/20.28'' Sleeve: 59.5cm/23.43'' Length: 65cm/25.59'' Size: XXL US:
        16 EU: 46 Bust: 112cm/44.09'' Shoulder: 53.5cm/21.06'' Sleeve:
        61.5cm/24.21'' Length: 67cm/26.38'' Size: XXXL US: 18 EU: 48 Bust:
        116cm/45.67'' Shoulder: 55.5cm/21.85'' Sleeve: 63.5cm/25.00'' Length:
        69cm/27.17'' cardigan sweaters for women sweaters for women womens
        sweaters oversized sweaters for women fall sweaters for women crew neck
        sweatershirt women white sweaters for women long sweaters for women
        sweaters aesthetic cute sweaters womens cardigan sweaters plus size
        sweaters for women girls sweaters black sweaters for women christmas
        sweaters for women halloween sweaters for women sweaters for women
        cardigan open front crewneck sweatshirts sweatshirts for teen girls
        womens sweatshirts graphic sweatshirts cute sweatshirts hoodies for teen
        girls hoodies aesthetic hoodies for girls hoodies for teens womens
        hoodies pullover womens hoodies zip up crop hoodies for women tie dye
        hoodies for women zip up hoodie hoodies for teen girls zip up hoodie
        women oversized hoodie brown zip up hoodie cropped zip up hoodie black
        zip up hoodie womens zip up hoodie work blouses for women office plus
        size blouses for women black blouses for women white blouses for women
        dressy women shirts and blouses blouses for women plus size"]
pipeline_tag: sentence-similarity
library_name: sentence-transformers

SentenceTransformer based on sentence-transformers/all-mpnet-base-v2

This is a sentence-transformers model finetuned from sentence-transformers/all-mpnet-base-v2. 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.

Model Details

Model Description

  • Model Type: Sentence Transformer
  • Base model: sentence-transformers/all-mpnet-base-v2
  • Maximum Sequence Length: 128 tokens
  • Output Dimensionality: 768 tokens
  • Similarity Function: Cosine Similarity

Model Sources

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 128, 'do_lower_case': False}) with Transformer model: MPNetModel 
  (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})
)

Usage

Direct Usage (Sentence Transformers)

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("knguyennguyen/mpnet_jacket4k")
# Run inference
sentences = [
    'a pair of statement stud earrings for brightening up outfits',
    "Title: Kendra Scott Sienna Ear Jacket Earrings Descripion: ['The Kendra Scott Sienna Ear Jacket Earrings are the best bright-colored statement stud. Straight post backing. Removeable jackets. Gold plated brass. Shell detailing. Remove jewelry when swimming, bathing, or exercising. Imported.']",
    "Title: TrailCrest Infant-Toddler Boys & Girls Fleece Full Zip Mock Neck Soft Jacket Descripion: ['Feel the chill! On chilly summer nights out in the wilderness or crisp spring and autumn days, these semi-fitted fleece zips are the perfect cover to keep you warm. These fleece zips are great for camping, hiking and just about any sport related activities. Featuring bright accent colors with realistic camo print to make the experience all the more exciting. Fully equipped with a mock neck collar for extra warmth, two hand pockets with hidden zippers, an adjustable drawstring at the bottom, and elastic cuffs. All set for the outdoors!']",
]
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]

Training Details

Training Dataset

Unnamed Dataset

  • Size: 7,598 training samples
  • Columns: sentence_0 and sentence_1
  • Approximate statistics based on the first 1000 samples:
    sentence_0 sentence_1
    type string string
    details
    • min: 4 tokens
    • mean: 18.2 tokens
    • max: 51 tokens
    • min: 35 tokens
    • mean: 105.77 tokens
    • max: 128 tokens
  • Samples:
    sentence_0 sentence_1
    men's parka jacket with a concealed front closure, multiple storage pockets, and a detachable hood. Title: DKNY Men's Welded Short Parka Jacket Descripion: ["FUNCTIONALITY: Center front placket with hidden zipper closure, two lower flap pockets with side entry, removable hood. STYLISH FEATURES: DKNY logo on wearer's left sleeve, DKNY small logo metal zipper puller. Water resistant outer shell with faux down fill insulation."]
    a track jacket and pant set for active girls Title: PUMA girls Track Jacket & Pant Set Descripion: ['PUMA, a Global athletic brand, provides consumers with innovative products that successfully fuses the creative influences from the world of sport, lifestyle, and fashion.']
    kids' hoodies made from soft fleece material, featuring a zip closure and pockets for convenience. suitable for both boys and girls, these garments are designed for warmth and comfort during colder seasons. Title: Eddie Bauer Kids’ Jacket - 2 Pack Ultra Soft Sherpa Fleece Hoodie Sweatshirt for Boys and Girls (5-20) Descripion: ["Eddie Bauer Kids' Plush Sherpa Fleece Zip Hoodie Sweatshirt for Boys and Girls is a great choice for the chilly fall and winter weather. This fun sweatshirt is high quality, and long lasting. This everyday clothing is a great gift for birthdays and holidays. We have several designs available, perfect for cold weather. High Quality fabric and durable, high stitch density give your boy or girl a long lasting hoodie that they'll be able to wear for years. Fashionable sherpa fleece sweatshirt for a soft feel and comfort. Full Zip jacket and quarter zip pullover allows her to have the ability to layer other tops underneath for a casual look. Comfort Fit jacket has split kangaroo pockets and cuffed sleeves to keep it from bunching and riding up. Simply machine wash and tumble dry; Please Reference the Variations for All Available Sizes & Colors! Eddie Bauer Offers Premium Clothing at Affordable Prices because we value every customer that visits our listings! Stop by Our Storefront to See the Rest of Our Great Deals; we're confident you're going to find items that anyone who needs a gift will absolutely love and adore!"]
  • Loss: MultipleNegativesRankingLoss with these parameters:
    {
        "scale": 20.0,
        "similarity_fct": "cos_sim"
    }
    

Training Hyperparameters

Non-Default Hyperparameters

  • per_device_train_batch_size: 128
  • per_device_eval_batch_size: 128
  • num_train_epochs: 5
  • multi_dataset_batch_sampler: round_robin

All Hyperparameters

Click to expand
  • overwrite_output_dir: False
  • do_predict: False
  • eval_strategy: no
  • prediction_loss_only: True
  • per_device_train_batch_size: 128
  • per_device_eval_batch_size: 128
  • per_gpu_train_batch_size: None
  • per_gpu_eval_batch_size: None
  • gradient_accumulation_steps: 1
  • eval_accumulation_steps: None
  • torch_empty_cache_steps: None
  • learning_rate: 5e-05
  • weight_decay: 0.0
  • adam_beta1: 0.9
  • adam_beta2: 0.999
  • adam_epsilon: 1e-08
  • max_grad_norm: 1
  • num_train_epochs: 5
  • max_steps: -1
  • lr_scheduler_type: linear
  • lr_scheduler_kwargs: {}
  • warmup_ratio: 0.0
  • warmup_steps: 0
  • log_level: passive
  • log_level_replica: warning
  • log_on_each_node: True
  • logging_nan_inf_filter: True
  • save_safetensors: True
  • save_on_each_node: False
  • save_only_model: False
  • restore_callback_states_from_checkpoint: False
  • no_cuda: False
  • use_cpu: False
  • use_mps_device: False
  • seed: 42
  • data_seed: None
  • jit_mode_eval: False
  • use_ipex: False
  • bf16: False
  • fp16: False
  • fp16_opt_level: O1
  • half_precision_backend: auto
  • bf16_full_eval: False
  • fp16_full_eval: False
  • tf32: None
  • local_rank: 0
  • ddp_backend: None
  • tpu_num_cores: None
  • tpu_metrics_debug: False
  • debug: []
  • dataloader_drop_last: False
  • dataloader_num_workers: 0
  • dataloader_prefetch_factor: None
  • past_index: -1
  • disable_tqdm: False
  • remove_unused_columns: True
  • label_names: None
  • load_best_model_at_end: False
  • ignore_data_skip: False
  • fsdp: []
  • fsdp_min_num_params: 0
  • fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
  • fsdp_transformer_layer_cls_to_wrap: None
  • accelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
  • deepspeed: None
  • label_smoothing_factor: 0.0
  • optim: adamw_torch
  • optim_args: None
  • adafactor: False
  • group_by_length: False
  • length_column_name: length
  • ddp_find_unused_parameters: None
  • ddp_bucket_cap_mb: None
  • ddp_broadcast_buffers: False
  • dataloader_pin_memory: True
  • dataloader_persistent_workers: False
  • skip_memory_metrics: True
  • use_legacy_prediction_loop: False
  • push_to_hub: False
  • resume_from_checkpoint: None
  • hub_model_id: None
  • hub_strategy: every_save
  • hub_private_repo: False
  • hub_always_push: False
  • gradient_checkpointing: False
  • gradient_checkpointing_kwargs: None
  • include_inputs_for_metrics: False
  • eval_do_concat_batches: True
  • fp16_backend: auto
  • push_to_hub_model_id: None
  • push_to_hub_organization: None
  • mp_parameters:
  • auto_find_batch_size: False
  • full_determinism: False
  • torchdynamo: None
  • ray_scope: last
  • ddp_timeout: 1800
  • torch_compile: False
  • torch_compile_backend: None
  • torch_compile_mode: None
  • dispatch_batches: None
  • split_batches: None
  • include_tokens_per_second: False
  • include_num_input_tokens_seen: False
  • neftune_noise_alpha: None
  • optim_target_modules: None
  • batch_eval_metrics: False
  • eval_on_start: False
  • use_liger_kernel: False
  • eval_use_gather_object: False
  • batch_sampler: batch_sampler
  • multi_dataset_batch_sampler: round_robin

Framework Versions

  • Python: 3.11.11
  • Sentence Transformers: 3.1.1
  • Transformers: 4.45.2
  • PyTorch: 2.5.1+cu121
  • Accelerate: 1.2.1
  • Datasets: 3.2.0
  • Tokenizers: 0.20.3

Citation

BibTeX

Sentence Transformers

@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",
}

MultipleNegativesRankingLoss

@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}
}