sentence-transformers
Safetensors
bert
retrieval
movie-recommendation
semantic-search
Eval Results (legacy)
Instructions to use JJTsao/fine-tuned_movie_retriever-all-minilm-l6-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use JJTsao/fine-tuned_movie_retriever-all-minilm-l6-v2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("JJTsao/fine-tuned_movie_retriever-all-minilm-l6-v2") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files- 1_Pooling/config.json +10 -0
- README.md +622 -0
- config.json +25 -0
- config_sentence_transformers.json +10 -0
- model.safetensors +3 -0
- modules.json +20 -0
- sentence_bert_config.json +4 -0
- special_tokens_map.json +37 -0
- tokenizer.json +0 -0
- tokenizer_config.json +65 -0
- vocab.txt +0 -0
1_Pooling/config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"word_embedding_dimension": 384,
|
| 3 |
+
"pooling_mode_cls_token": false,
|
| 4 |
+
"pooling_mode_mean_tokens": true,
|
| 5 |
+
"pooling_mode_max_tokens": false,
|
| 6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
+
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
+
"pooling_mode_lasttoken": false,
|
| 9 |
+
"include_prompt": true
|
| 10 |
+
}
|
README.md
ADDED
|
@@ -0,0 +1,622 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- sentence-transformers
|
| 4 |
+
- sentence-similarity
|
| 5 |
+
- feature-extraction
|
| 6 |
+
- generated_from_trainer
|
| 7 |
+
- dataset_size:32382
|
| 8 |
+
- loss:MultipleNegativesRankingLoss
|
| 9 |
+
base_model: sentence-transformers/all-MiniLM-L6-v2
|
| 10 |
+
widget:
|
| 11 |
+
- source_sentence: Can you recommend something where Katie Douglas struggles with
|
| 12 |
+
dystopia and girls' boarding school
|
| 13 |
+
sentences:
|
| 14 |
+
- 'Title: Freeheld
|
| 15 |
+
|
| 16 |
+
Genres: Romance, Drama
|
| 17 |
+
|
| 18 |
+
Overview: New Jersey car mechanic Stacie Andree and her police detective girlfriend
|
| 19 |
+
Laurel Hester both battle to secure Hester''s pension benefits after she was diagnosed
|
| 20 |
+
with a terminal illness.
|
| 21 |
+
|
| 22 |
+
Tagline: A true story of love and injustice.
|
| 23 |
+
|
| 24 |
+
Director: Peter Sollett
|
| 25 |
+
|
| 26 |
+
Stars: Julianne Moore, Elliot Page, Steve Carell
|
| 27 |
+
|
| 28 |
+
Release Date: 2015-10-02
|
| 29 |
+
|
| 30 |
+
Keywords: new jersey, equality, lesbian relationship, cancer, lesbian, assertive'
|
| 31 |
+
- 'Title: Level 16
|
| 32 |
+
|
| 33 |
+
Genres: Science Fiction, Thriller
|
| 34 |
+
|
| 35 |
+
Overview: The teenage girls of Vestalis Academy are meticulously trained in the
|
| 36 |
+
art of being “clean girls,” practicing the virtues of perfect femininity. But
|
| 37 |
+
what exactly are they being trained for? Vivien intends to find out.
|
| 38 |
+
|
| 39 |
+
Tagline: Purity has a price.
|
| 40 |
+
|
| 41 |
+
Director: Danishka Esterhazy
|
| 42 |
+
|
| 43 |
+
Stars: Katie Douglas, Celina Martin, Peter Outerbridge
|
| 44 |
+
|
| 45 |
+
Release Date: 2018-02-20
|
| 46 |
+
|
| 47 |
+
Keywords: truth, orphanage, dystopia, imprisonment, girls'' boarding school, betrayal,
|
| 48 |
+
woman director, dark secret, mysterious, girls'' school'
|
| 49 |
+
- "Title: Believe Me: The Abduction of Lisa McVey\nGenres: Drama, TV Movie, Crime\n\
|
| 50 |
+
Overview: On the night she plans on taking her own life, 17-year-old 'Lisa McVey'\
|
| 51 |
+
\ is kidnapped and finds herself fighting to stay alive and manages to be a victim\
|
| 52 |
+
\ of rape. She manages to talk her attacker into releasing her, but when she returns\
|
| 53 |
+
\ home, no one believes her story except for one detective, who suspects she was\
|
| 54 |
+
\ abducted by a serial killer. Based on horrifying true events.\nTagline: \nDirector:\
|
| 55 |
+
\ Jim Donovan\nStars: Katie Douglas, David James Elliott, Rossif Sutherland\n\
|
| 56 |
+
Release Date: 2018-09-30\nKeywords: suicide, sexual abuse, kidnapping, detective,\
|
| 57 |
+
\ based on true story, serial killer, murderer, held captive, sole survivor, police\
|
| 58 |
+
\ force, sexual assault, donut shop, abduction, re-creation, teenager, survivor"
|
| 59 |
+
- source_sentence: Films about friendship and freedom like The One and Only Ivan
|
| 60 |
+
sentences:
|
| 61 |
+
- 'Title: The One and Only Ivan
|
| 62 |
+
|
| 63 |
+
Genres: Family, Comedy, Drama
|
| 64 |
+
|
| 65 |
+
Overview: A gorilla named Ivan who’s living in a suburban shopping mall tries
|
| 66 |
+
to piece together his past, with the help of other animals, as they hatch a plan
|
| 67 |
+
to escape from captivity.
|
| 68 |
+
|
| 69 |
+
Tagline: It''s never too late to discover who you are.
|
| 70 |
+
|
| 71 |
+
Director: Thea Sharrock
|
| 72 |
+
|
| 73 |
+
Stars: Bryan Cranston, Sam Rockwell, Brooklynn Prince
|
| 74 |
+
|
| 75 |
+
Release Date: 2020-08-21
|
| 76 |
+
|
| 77 |
+
Keywords: based on novel or book, circus, gorilla, based on true story, talking
|
| 78 |
+
animal'
|
| 79 |
+
- "Title: We Are the World: The Story Behind the Song\nGenres: Music, Documentary\n\
|
| 80 |
+
Overview: We Are the World: The Story Behind the Song is a documentary which examines\
|
| 81 |
+
\ how the song was written, how producer Quincy Jones and songwriters Michael\
|
| 82 |
+
\ Jackson and Lionel Richie persuaded some of the most popular performers in America\
|
| 83 |
+
\ to donate their services to the project, and offers a behind-the-scenes look\
|
| 84 |
+
\ at the marathon recording session that produced the single.\nTagline: \nDirector:\
|
| 85 |
+
\ Tom Trbovich\nStars: Jane Fonda, Quincy Jones, Bob Dylan\nRelease Date: 1985-01-28\n\
|
| 86 |
+
Keywords: charity, behind the scenes, singing"
|
| 87 |
+
- 'Title: Temptation: Confessions of a Marriage Counselor
|
| 88 |
+
|
| 89 |
+
Genres: Drama
|
| 90 |
+
|
| 91 |
+
Overview: Judith has known her husband Brice since they were children, but now
|
| 92 |
+
their marriage is growing stale. Having just completed her graduate work in psychotherapy,
|
| 93 |
+
she''s eager to begin a career as a marriage counselor. She takes an internship
|
| 94 |
+
at a matchmaking firm for millionaires and meets Harley, a charismatic billionaire
|
| 95 |
+
investor who makes no effort to hide his attraction to Judith. Although quite
|
| 96 |
+
resistant at first, eventually Judith succumbs to his charms, placing her marriage
|
| 97 |
+
in jeopardy and forever altering the course of her life.
|
| 98 |
+
|
| 99 |
+
Tagline: Betrayal burns forever.
|
| 100 |
+
|
| 101 |
+
Director: Tyler Perry
|
| 102 |
+
|
| 103 |
+
Stars: Jurnee Smollett, Lance Gross, Robbie Jones
|
| 104 |
+
|
| 105 |
+
Release Date: 2013-03-29
|
| 106 |
+
|
| 107 |
+
Keywords: husband wife relationship, washington dc, usa, therapist, marriage crisis,
|
| 108 |
+
married couple, pharmacist, new orleans, louisiana, betrayal, billionaire, temptation,
|
| 109 |
+
young love, marriage counselor, crisis of faith, test of faithfulness, young couple,
|
| 110 |
+
childhood sweethearts, black cinema, matchmaker, christian faith, financial investor,
|
| 111 |
+
matchmaking service'
|
| 112 |
+
- source_sentence: Memorable adventure films focused on super power and grappling
|
| 113 |
+
with based on comic
|
| 114 |
+
sentences:
|
| 115 |
+
- 'Title: Green Lantern: Emerald Knights
|
| 116 |
+
|
| 117 |
+
Genres: Action, Adventure, Animation, Science Fiction, Drama
|
| 118 |
+
|
| 119 |
+
Overview: As the home planet of the Green Lantern Corps faces a battle with an
|
| 120 |
+
ancient enemy, Hal Jordan prepares new recruit Arisia for the coming conflict
|
| 121 |
+
by relating stories of the first Green Lantern and several of Hal''s comrades.
|
| 122 |
+
|
| 123 |
+
Tagline: Witness the Color of Will.
|
| 124 |
+
|
| 125 |
+
Director: Jay Oliva
|
| 126 |
+
|
| 127 |
+
Stars: Nathan Fillion, Jason Isaacs, Elisabeth Moss
|
| 128 |
+
|
| 129 |
+
Release Date: 2011-04-09
|
| 130 |
+
|
| 131 |
+
Keywords: superhero, based on comic, super power, woman director'
|
| 132 |
+
- 'Title: Monkey Man
|
| 133 |
+
|
| 134 |
+
Genres: Action, Thriller, Crime
|
| 135 |
+
|
| 136 |
+
Overview: Kid is an anonymous young man who ekes out a meager living in an underground
|
| 137 |
+
fight club where, night after night, wearing a gorilla mask, he is beaten bloody
|
| 138 |
+
by more popular fighters for cash. After years of suppressed rage, Kid discovers
|
| 139 |
+
a way to infiltrate the enclave of the city’s sinister elite. As his childhood
|
| 140 |
+
trauma boils over, his mysteriously scarred hands unleash an explosive campaign
|
| 141 |
+
of retribution to settle the score with the men who took everything from him.
|
| 142 |
+
|
| 143 |
+
Tagline: One small ember can burn down everything.
|
| 144 |
+
|
| 145 |
+
Director: Dev Patel
|
| 146 |
+
|
| 147 |
+
Stars: Dev Patel, Sikandar Kher, Makrand Deshpande
|
| 148 |
+
|
| 149 |
+
Release Date: 2024-04-03
|
| 150 |
+
|
| 151 |
+
Keywords: indian, revenge thriller, action thriller, night club, трансгендер,
|
| 152 |
+
лгбт'
|
| 153 |
+
- 'Title: The Watchers
|
| 154 |
+
|
| 155 |
+
Genres: Fantasy, Horror, Mystery
|
| 156 |
+
|
| 157 |
+
Overview: A young artist gets stranded in an extensive, immaculate forest in western
|
| 158 |
+
Ireland, where, after finding shelter, she becomes trapped alongside three strangers,
|
| 159 |
+
stalked by mysterious creatures each night.
|
| 160 |
+
|
| 161 |
+
Tagline: Let them see you.
|
| 162 |
+
|
| 163 |
+
Director: Ishana Night Shyamalan
|
| 164 |
+
|
| 165 |
+
Stars: Dakota Fanning, Georgina Campbell, Olwen Fouéré
|
| 166 |
+
|
| 167 |
+
Release Date: 2024-06-06
|
| 168 |
+
|
| 169 |
+
Keywords: based on novel or book, shapeshifting, bunker, forest, shelter, ireland,
|
| 170 |
+
changeling, strangers, mimic, fairies, complicated, trapped in a room, irish mythology'
|
| 171 |
+
- source_sentence: Any recommendations for top drama films from 2000 featuring Claire
|
| 172 |
+
Danes?
|
| 173 |
+
sentences:
|
| 174 |
+
- 'Title: Igby Goes Down
|
| 175 |
+
|
| 176 |
+
Genres: Comedy, Drama
|
| 177 |
+
|
| 178 |
+
Overview: Igby Slocumb, a rebellious and sarcastic 17-year-old boy, is at war
|
| 179 |
+
with the stifling world of old money privilege he was born into. With a schizophrenic
|
| 180 |
+
father, a self-absorbed, distant mother, and a shark-like young Republican big
|
| 181 |
+
brother, Igby figures there must be a better life out there -- and sets about
|
| 182 |
+
finding it.
|
| 183 |
+
|
| 184 |
+
Tagline: Insanity is relative.
|
| 185 |
+
|
| 186 |
+
Director: Burr Steers
|
| 187 |
+
|
| 188 |
+
Stars: Kieran Culkin, Claire Danes, Jeff Goldblum
|
| 189 |
+
|
| 190 |
+
Release Date: 2002-09-13
|
| 191 |
+
|
| 192 |
+
Keywords: new york city, parent child relationship, falling in love, godfather'
|
| 193 |
+
- 'Title: Significant Other
|
| 194 |
+
|
| 195 |
+
Genres: Science Fiction, Thriller, Horror, Mystery, Drama
|
| 196 |
+
|
| 197 |
+
Overview: Ruth and Harry decide to take a romantic backpacking trip through the
|
| 198 |
+
Pacific Northwest, but amongst the beautiful scenery, Ruth makes an unexpected
|
| 199 |
+
discovery that sets her off on a strange, frightening new path. The couple aren''t
|
| 200 |
+
alone in the woods, and they might not be the same when they come out...if they
|
| 201 |
+
come out.
|
| 202 |
+
|
| 203 |
+
Tagline: Do you really know the one you love?
|
| 204 |
+
|
| 205 |
+
Director: Robert Olsen
|
| 206 |
+
|
| 207 |
+
Stars: Maika Monroe, Jake Lacy, Matthew Yang King
|
| 208 |
+
|
| 209 |
+
Release Date: 2022-10-06
|
| 210 |
+
|
| 211 |
+
Keywords: marriage proposal, camping, cliff, hiking, wilderness, forest, alien
|
| 212 |
+
life-form, deer, couple, alien possession, anxiety attack, anxiety disorder, pacific
|
| 213 |
+
northwest'
|
| 214 |
+
- 'Title: U-571
|
| 215 |
+
|
| 216 |
+
Genres: War, Action, Adventure
|
| 217 |
+
|
| 218 |
+
Overview: In the midst of World War II, the battle under the sea rages and the
|
| 219 |
+
Nazis have the upper hand as the Allies are unable to crack their war codes. However,
|
| 220 |
+
after a wrecked U-boat sends out an SOS signal, the Allies realise this is their
|
| 221 |
+
chance to seize the ''enigma coding machine''.
|
| 222 |
+
|
| 223 |
+
Tagline: Heroes are ordinary men who do extraordinary things in extraordinary
|
| 224 |
+
times.
|
| 225 |
+
|
| 226 |
+
Director: Jonathan Mostow
|
| 227 |
+
|
| 228 |
+
Stars: Matthew McConaughey, Bill Paxton, Harvey Keitel
|
| 229 |
+
|
| 230 |
+
Release Date: 2000-04-20
|
| 231 |
+
|
| 232 |
+
Keywords: mission, submarine, world war ii, north atlantic, enigma machine, adoring'
|
| 233 |
+
- source_sentence: Action-packed comedy with a buddy cop twist and high school shenanigans
|
| 234 |
+
sentences:
|
| 235 |
+
- 'Title: 21 Jump Street
|
| 236 |
+
|
| 237 |
+
Genres: Action, Comedy, Crime
|
| 238 |
+
|
| 239 |
+
Overview: When cops Schmidt and Jenko join the secret Jump Street unit, they use
|
| 240 |
+
their youthful appearances to go undercover as high school students. They trade
|
| 241 |
+
in their guns and badges for backpacks, and set out to shut down a dangerous drug
|
| 242 |
+
ring. But, as time goes on, Schmidt and Jenko discover that high school is nothing
|
| 243 |
+
like it was just a few years earlier -- and, what''s more, they must again confront
|
| 244 |
+
the teenage terror and anxiety they thought they had left behind.
|
| 245 |
+
|
| 246 |
+
Tagline: The only thing getting blown tonight is their cover.
|
| 247 |
+
|
| 248 |
+
Director: Phil Lord
|
| 249 |
+
|
| 250 |
+
Stars: Jonah Hill, Channing Tatum, Brie Larson
|
| 251 |
+
|
| 252 |
+
Release Date: 2012-03-14
|
| 253 |
+
|
| 254 |
+
Keywords: drug dealer, high school, investigation, nerd, male friendship, parody,
|
| 255 |
+
jock, crude humor, undercover cop, high school student, police academy, buddy
|
| 256 |
+
cop, buddy comedy, duringcreditsstinger, high school rivalry, bicycle cop, based
|
| 257 |
+
on tv series, cops, party drugs, high school classmates'
|
| 258 |
+
- 'Title: Free Fire
|
| 259 |
+
|
| 260 |
+
Genres: Action, Crime, Mystery
|
| 261 |
+
|
| 262 |
+
Overview: Set in Boston in 1978, a meeting in a deserted warehouse between two
|
| 263 |
+
gangs turns into a shoot-out and a game of survival.
|
| 264 |
+
|
| 265 |
+
Tagline: All guns. No control.
|
| 266 |
+
|
| 267 |
+
Director: Ben Wheatley
|
| 268 |
+
|
| 269 |
+
Stars: Cillian Murphy, Brie Larson, Armie Hammer
|
| 270 |
+
|
| 271 |
+
Release Date: 2017-03-31
|
| 272 |
+
|
| 273 |
+
Keywords: sniper, boston, massachusetts, 1970s, gang, shootout, warehouse, mischievous,
|
| 274 |
+
tense, amused'
|
| 275 |
+
- 'Title: The Bodyguard
|
| 276 |
+
|
| 277 |
+
Genres: Action, Drama, Romance
|
| 278 |
+
|
| 279 |
+
Overview: A former Secret Service agent grudgingly takes an assignment to protect
|
| 280 |
+
a pop idol who''s threatened by a crazed fan. At first, the safety-obsessed bodyguard
|
| 281 |
+
and the self-indulgent diva totally clash. But before long, all that tension sparks
|
| 282 |
+
fireworks of another sort, and the love-averse tough guy is torn between duty
|
| 283 |
+
and romance.
|
| 284 |
+
|
| 285 |
+
Tagline: Never let her out of your sight. Never let your guard down. Never fall
|
| 286 |
+
in love.
|
| 287 |
+
|
| 288 |
+
Director: Mick Jackson
|
| 289 |
+
|
| 290 |
+
Stars: Kevin Costner, Whitney Houston, Gary Kemp
|
| 291 |
+
|
| 292 |
+
Release Date: 1992-11-25
|
| 293 |
+
|
| 294 |
+
Keywords: sibling relationship, bodyguard, anonymous letter, diva, stalker, singer,
|
| 295 |
+
los angeles, california, admiring'
|
| 296 |
+
pipeline_tag: sentence-similarity
|
| 297 |
+
library_name: sentence-transformers
|
| 298 |
+
---
|
| 299 |
+
|
| 300 |
+
# SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
|
| 301 |
+
|
| 302 |
+
This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2). It maps sentences & paragraphs to a 384-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
|
| 303 |
+
|
| 304 |
+
## Model Details
|
| 305 |
+
|
| 306 |
+
### Model Description
|
| 307 |
+
- **Model Type:** Sentence Transformer
|
| 308 |
+
- **Base model:** [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) <!-- at revision c9745ed1d9f207416be6d2e6f8de32d1f16199bf -->
|
| 309 |
+
- **Maximum Sequence Length:** 256 tokens
|
| 310 |
+
- **Output Dimensionality:** 384 dimensions
|
| 311 |
+
- **Similarity Function:** Cosine Similarity
|
| 312 |
+
<!-- - **Training Dataset:** Unknown -->
|
| 313 |
+
<!-- - **Language:** Unknown -->
|
| 314 |
+
<!-- - **License:** Unknown -->
|
| 315 |
+
|
| 316 |
+
### Model Sources
|
| 317 |
+
|
| 318 |
+
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
| 319 |
+
- **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
|
| 320 |
+
- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
| 321 |
+
|
| 322 |
+
### Full Model Architecture
|
| 323 |
+
|
| 324 |
+
```
|
| 325 |
+
SentenceTransformer(
|
| 326 |
+
(0): Transformer({'max_seq_length': 256, 'do_lower_case': False}) with Transformer model: BertModel
|
| 327 |
+
(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})
|
| 328 |
+
(2): Normalize()
|
| 329 |
+
)
|
| 330 |
+
```
|
| 331 |
+
|
| 332 |
+
## Usage
|
| 333 |
+
|
| 334 |
+
### Direct Usage (Sentence Transformers)
|
| 335 |
+
|
| 336 |
+
First install the Sentence Transformers library:
|
| 337 |
+
|
| 338 |
+
```bash
|
| 339 |
+
pip install -U sentence-transformers
|
| 340 |
+
```
|
| 341 |
+
|
| 342 |
+
Then you can load this model and run inference.
|
| 343 |
+
```python
|
| 344 |
+
from sentence_transformers import SentenceTransformer
|
| 345 |
+
|
| 346 |
+
# Download from the 🤗 Hub
|
| 347 |
+
model = SentenceTransformer("sentence_transformers_model_id")
|
| 348 |
+
# Run inference
|
| 349 |
+
sentences = [
|
| 350 |
+
'Action-packed comedy with a buddy cop twist and high school shenanigans',
|
| 351 |
+
"Title: 21 Jump Street\nGenres: Action, Comedy, Crime\nOverview: When cops Schmidt and Jenko join the secret Jump Street unit, they use their youthful appearances to go undercover as high school students. They trade in their guns and badges for backpacks, and set out to shut down a dangerous drug ring. But, as time goes on, Schmidt and Jenko discover that high school is nothing like it was just a few years earlier -- and, what's more, they must again confront the teenage terror and anxiety they thought they had left behind.\nTagline: The only thing getting blown tonight is their cover.\nDirector: Phil Lord\nStars: Jonah Hill, Channing Tatum, Brie Larson\nRelease Date: 2012-03-14\nKeywords: drug dealer, high school, investigation, nerd, male friendship, parody, jock, crude humor, undercover cop, high school student, police academy, buddy cop, buddy comedy, duringcreditsstinger, high school rivalry, bicycle cop, based on tv series, cops, party drugs, high school classmates",
|
| 352 |
+
'Title: Free Fire\nGenres: Action, Crime, Mystery\nOverview: Set in Boston in 1978, a meeting in a deserted warehouse between two gangs turns into a shoot-out and a game of survival.\nTagline: All guns. No control.\nDirector: Ben Wheatley\nStars: Cillian Murphy, Brie Larson, Armie Hammer\nRelease Date: 2017-03-31\nKeywords: sniper, boston, massachusetts, 1970s, gang, shootout, warehouse, mischievous, tense, amused',
|
| 353 |
+
]
|
| 354 |
+
embeddings = model.encode(sentences)
|
| 355 |
+
print(embeddings.shape)
|
| 356 |
+
# [3, 384]
|
| 357 |
+
|
| 358 |
+
# Get the similarity scores for the embeddings
|
| 359 |
+
similarities = model.similarity(embeddings, embeddings)
|
| 360 |
+
print(similarities.shape)
|
| 361 |
+
# [3, 3]
|
| 362 |
+
```
|
| 363 |
+
|
| 364 |
+
<!--
|
| 365 |
+
### Direct Usage (Transformers)
|
| 366 |
+
|
| 367 |
+
<details><summary>Click to see the direct usage in Transformers</summary>
|
| 368 |
+
|
| 369 |
+
</details>
|
| 370 |
+
-->
|
| 371 |
+
|
| 372 |
+
<!--
|
| 373 |
+
### Downstream Usage (Sentence Transformers)
|
| 374 |
+
|
| 375 |
+
You can finetune this model on your own dataset.
|
| 376 |
+
|
| 377 |
+
<details><summary>Click to expand</summary>
|
| 378 |
+
|
| 379 |
+
</details>
|
| 380 |
+
-->
|
| 381 |
+
|
| 382 |
+
<!--
|
| 383 |
+
### Out-of-Scope Use
|
| 384 |
+
|
| 385 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 386 |
+
-->
|
| 387 |
+
|
| 388 |
+
<!--
|
| 389 |
+
## Bias, Risks and Limitations
|
| 390 |
+
|
| 391 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| 392 |
+
-->
|
| 393 |
+
|
| 394 |
+
<!--
|
| 395 |
+
### Recommendations
|
| 396 |
+
|
| 397 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| 398 |
+
-->
|
| 399 |
+
|
| 400 |
+
## Training Details
|
| 401 |
+
|
| 402 |
+
### Training Dataset
|
| 403 |
+
|
| 404 |
+
#### Unnamed Dataset
|
| 405 |
+
|
| 406 |
+
* Size: 32,382 training samples
|
| 407 |
+
* Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>sentence_2</code>
|
| 408 |
+
* Approximate statistics based on the first 1000 samples:
|
| 409 |
+
| | sentence_0 | sentence_1 | sentence_2 |
|
| 410 |
+
|:--------|:---------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|
|
| 411 |
+
| type | string | string | string |
|
| 412 |
+
| details | <ul><li>min: 7 tokens</li><li>mean: 15.9 tokens</li><li>max: 31 tokens</li></ul> | <ul><li>min: 54 tokens</li><li>mean: 150.05 tokens</li><li>max: 256 tokens</li></ul> | <ul><li>min: 70 tokens</li><li>mean: 147.88 tokens</li><li>max: 256 tokens</li></ul> |
|
| 413 |
+
* Samples:
|
| 414 |
+
| sentence_0 | sentence_1 | sentence_2 |
|
| 415 |
+
|:----------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
| 416 |
+
| <code>Romantic comedy with meddling family dynamics and hilarious schemes</code> | <code>Title: Monster-in-Law<br>Genres: Romance, Comedy<br>Overview: Office temp Charlotte Cantilini thinks she's found Mr. Right when she starts dating gorgeous surgeon Dr. Kevin Fields. But there's a problem standing in the way of everlasting bliss: Kevin's overbearing and controlling mother, Viola. Fearing she'll lose her son's affections forever, Viola decides to break up the happy couple by becoming the world's worst mother-in-law.<br>Tagline: This relationship is going to be a real mother.<br>Director: Robert Luketic<br>Stars: Jennifer Lopez, Jane Fonda, Michael Vartan<br>Release Date: 2005-05-13<br>Keywords: surgeon, man of one's dreams, free spirit, falling in love, overbearing mother, drinking, nervous breakdown, fiancé fiancée relationship, fashion design, rich man poor woman, scheme, wealth differences, rich family, imminent wedding, mother son relationship, interfering mother-in-law, disapproving parent, talk show host, best friends, controlling parent, meddling mother, moving in with significant othe...</code> | <code>Title: Bad Sister<br>Genres: Thriller, TV Movie<br>Overview: As a top student at St. Adeline's Catholic Boarding School, Zoe senses that something is not quite right about the school's new nun-- a sense proven to be true when it is revealed the "good' nun is an imposter with a fatal attraction to Zoe's brother.<br>Tagline: How long has it been since your last confession?<br>Director: Doug Campbell<br>Stars: Alyshia Ochse, Devon Werkheiser, Ryan Whitney<br>Release Date: 2015-08-24<br>Keywords: erotic thriller</code> |
|
| 417 |
+
| <code>Movies with a humorous superhero vibe and adventurous alternate realities.</code> | <code>Title: Henry Danger: The Movie<br>Genres: Action, Comedy, Family, Science Fiction<br>Overview: Henry Hart meets a superfan—eager to fight crime with Kid Danger—who comes into possession of a device that can open up alternate realities. Facing a wild ride, Henry will need his best friend Jasper and his new superfan sidekick to find his way out or be stuck in another dimension forever.<br>Tagline: Danger... Like you've never seen before.<br>Director: Joe Menendez<br>Stars: Jace Norman, Sean Ryan Fox, Ella Anderson<br>Release Date: 2025-01-17<br>Keywords: superhero, supervillain, based on tv series</code> | <code>Title: Inuyasha the Movie 3: Swords of an Honorable Ruler<br>Genres: Animation, Fantasy, Adventure<br>Overview: Inuyasha and his brother, Sesshomaru, each inherited a sword from their father after his death. However, their father had a third sword, named Sounga, that he sealed away. Seven hundreds years after his death, Sounga awakens and threatens mankind's very existence. How will the children of the Great Dog Demon stop this unimaginable power?<br>Tagline: The Three Swords of the Fang collide in an epic battle!<br>Director: Toshiya Shinohara<br>Stars: Kappei Yamaguchi, Satsuki Yukino, Koji Tsujitani<br>Release Date: 2003-12-20<br>Keywords: youkai, shounen, anime</code> |
|
| 418 |
+
| <code>Heartfelt films about starting over and pursuing dreams</code> | <code>Title: 12th Fail<br>Genres: Drama<br>Overview: Based on the true story of IPS officer Manoj Kumar Sharma, 12th Fail sheds limelight on fearlessly embracing the idea of restarting the academic journey despite the setbacks and challenges and reclaiming one's destiny at a place where millions of students attempt the world's toughest competitive exam: UPSC.<br>Tagline: #RESTART<br>Director: Vidhu Vinod Chopra<br>Stars: Vikrant Massey, Medha Shankr, Anant Joshi<br>Release Date: 2023-08-11<br>Keywords: bollywood, awestruck</code> | <code>Title: Enchanted<br>Genres: Comedy, Family, Fantasy, Romance<br>Overview: The beautiful princess Giselle is banished by an evil queen from her magical, musical animated land and finds herself in the gritty reality of the streets of modern-day Manhattan. Shocked by this strange new environment that doesn't operate on a "happily ever after" basis, Giselle is now adrift in a chaotic world badly in need of enchantment. But when Giselle begins to fall in love with a charmingly flawed divorce lawyer who has come to her aid - even though she is already promised to a perfect fairy tale prince back home - she has to wonder: Can a storybook view of romance survive in the real world?<br>Tagline: This fairytale princess is about to meet a real Prince Charming.<br>Director: Kevin Lima<br>Stars: Amy Adams, Patrick Dempsey, James Marsden<br>Release Date: 2007-11-20<br>Keywords: princess, new york city, magic, fairy tale, poison, queen, prince, villain, musical, portal, female protagonist, female villain, manhattan, new y...</code> |
|
| 419 |
+
* Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
|
| 420 |
+
```json
|
| 421 |
+
{
|
| 422 |
+
"scale": 20.0,
|
| 423 |
+
"similarity_fct": "cos_sim"
|
| 424 |
+
}
|
| 425 |
+
```
|
| 426 |
+
|
| 427 |
+
### Training Hyperparameters
|
| 428 |
+
#### Non-Default Hyperparameters
|
| 429 |
+
|
| 430 |
+
- `per_device_train_batch_size`: 32
|
| 431 |
+
- `per_device_eval_batch_size`: 32
|
| 432 |
+
- `num_train_epochs`: 4
|
| 433 |
+
- `multi_dataset_batch_sampler`: round_robin
|
| 434 |
+
|
| 435 |
+
#### All Hyperparameters
|
| 436 |
+
<details><summary>Click to expand</summary>
|
| 437 |
+
|
| 438 |
+
- `overwrite_output_dir`: False
|
| 439 |
+
- `do_predict`: False
|
| 440 |
+
- `eval_strategy`: no
|
| 441 |
+
- `prediction_loss_only`: True
|
| 442 |
+
- `per_device_train_batch_size`: 32
|
| 443 |
+
- `per_device_eval_batch_size`: 32
|
| 444 |
+
- `per_gpu_train_batch_size`: None
|
| 445 |
+
- `per_gpu_eval_batch_size`: None
|
| 446 |
+
- `gradient_accumulation_steps`: 1
|
| 447 |
+
- `eval_accumulation_steps`: None
|
| 448 |
+
- `torch_empty_cache_steps`: None
|
| 449 |
+
- `learning_rate`: 5e-05
|
| 450 |
+
- `weight_decay`: 0.0
|
| 451 |
+
- `adam_beta1`: 0.9
|
| 452 |
+
- `adam_beta2`: 0.999
|
| 453 |
+
- `adam_epsilon`: 1e-08
|
| 454 |
+
- `max_grad_norm`: 1
|
| 455 |
+
- `num_train_epochs`: 4
|
| 456 |
+
- `max_steps`: -1
|
| 457 |
+
- `lr_scheduler_type`: linear
|
| 458 |
+
- `lr_scheduler_kwargs`: {}
|
| 459 |
+
- `warmup_ratio`: 0.0
|
| 460 |
+
- `warmup_steps`: 0
|
| 461 |
+
- `log_level`: passive
|
| 462 |
+
- `log_level_replica`: warning
|
| 463 |
+
- `log_on_each_node`: True
|
| 464 |
+
- `logging_nan_inf_filter`: True
|
| 465 |
+
- `save_safetensors`: True
|
| 466 |
+
- `save_on_each_node`: False
|
| 467 |
+
- `save_only_model`: False
|
| 468 |
+
- `restore_callback_states_from_checkpoint`: False
|
| 469 |
+
- `no_cuda`: False
|
| 470 |
+
- `use_cpu`: False
|
| 471 |
+
- `use_mps_device`: False
|
| 472 |
+
- `seed`: 42
|
| 473 |
+
- `data_seed`: None
|
| 474 |
+
- `jit_mode_eval`: False
|
| 475 |
+
- `use_ipex`: False
|
| 476 |
+
- `bf16`: False
|
| 477 |
+
- `fp16`: False
|
| 478 |
+
- `fp16_opt_level`: O1
|
| 479 |
+
- `half_precision_backend`: auto
|
| 480 |
+
- `bf16_full_eval`: False
|
| 481 |
+
- `fp16_full_eval`: False
|
| 482 |
+
- `tf32`: None
|
| 483 |
+
- `local_rank`: 0
|
| 484 |
+
- `ddp_backend`: None
|
| 485 |
+
- `tpu_num_cores`: None
|
| 486 |
+
- `tpu_metrics_debug`: False
|
| 487 |
+
- `debug`: []
|
| 488 |
+
- `dataloader_drop_last`: False
|
| 489 |
+
- `dataloader_num_workers`: 0
|
| 490 |
+
- `dataloader_prefetch_factor`: None
|
| 491 |
+
- `past_index`: -1
|
| 492 |
+
- `disable_tqdm`: False
|
| 493 |
+
- `remove_unused_columns`: True
|
| 494 |
+
- `label_names`: None
|
| 495 |
+
- `load_best_model_at_end`: False
|
| 496 |
+
- `ignore_data_skip`: False
|
| 497 |
+
- `fsdp`: []
|
| 498 |
+
- `fsdp_min_num_params`: 0
|
| 499 |
+
- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
|
| 500 |
+
- `tp_size`: 0
|
| 501 |
+
- `fsdp_transformer_layer_cls_to_wrap`: None
|
| 502 |
+
- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
|
| 503 |
+
- `deepspeed`: None
|
| 504 |
+
- `label_smoothing_factor`: 0.0
|
| 505 |
+
- `optim`: adamw_torch
|
| 506 |
+
- `optim_args`: None
|
| 507 |
+
- `adafactor`: False
|
| 508 |
+
- `group_by_length`: False
|
| 509 |
+
- `length_column_name`: length
|
| 510 |
+
- `ddp_find_unused_parameters`: None
|
| 511 |
+
- `ddp_bucket_cap_mb`: None
|
| 512 |
+
- `ddp_broadcast_buffers`: False
|
| 513 |
+
- `dataloader_pin_memory`: True
|
| 514 |
+
- `dataloader_persistent_workers`: False
|
| 515 |
+
- `skip_memory_metrics`: True
|
| 516 |
+
- `use_legacy_prediction_loop`: False
|
| 517 |
+
- `push_to_hub`: False
|
| 518 |
+
- `resume_from_checkpoint`: None
|
| 519 |
+
- `hub_model_id`: None
|
| 520 |
+
- `hub_strategy`: every_save
|
| 521 |
+
- `hub_private_repo`: None
|
| 522 |
+
- `hub_always_push`: False
|
| 523 |
+
- `gradient_checkpointing`: False
|
| 524 |
+
- `gradient_checkpointing_kwargs`: None
|
| 525 |
+
- `include_inputs_for_metrics`: False
|
| 526 |
+
- `include_for_metrics`: []
|
| 527 |
+
- `eval_do_concat_batches`: True
|
| 528 |
+
- `fp16_backend`: auto
|
| 529 |
+
- `push_to_hub_model_id`: None
|
| 530 |
+
- `push_to_hub_organization`: None
|
| 531 |
+
- `mp_parameters`:
|
| 532 |
+
- `auto_find_batch_size`: False
|
| 533 |
+
- `full_determinism`: False
|
| 534 |
+
- `torchdynamo`: None
|
| 535 |
+
- `ray_scope`: last
|
| 536 |
+
- `ddp_timeout`: 1800
|
| 537 |
+
- `torch_compile`: False
|
| 538 |
+
- `torch_compile_backend`: None
|
| 539 |
+
- `torch_compile_mode`: None
|
| 540 |
+
- `include_tokens_per_second`: False
|
| 541 |
+
- `include_num_input_tokens_seen`: False
|
| 542 |
+
- `neftune_noise_alpha`: None
|
| 543 |
+
- `optim_target_modules`: None
|
| 544 |
+
- `batch_eval_metrics`: False
|
| 545 |
+
- `eval_on_start`: False
|
| 546 |
+
- `use_liger_kernel`: False
|
| 547 |
+
- `eval_use_gather_object`: False
|
| 548 |
+
- `average_tokens_across_devices`: False
|
| 549 |
+
- `prompts`: None
|
| 550 |
+
- `batch_sampler`: batch_sampler
|
| 551 |
+
- `multi_dataset_batch_sampler`: round_robin
|
| 552 |
+
|
| 553 |
+
</details>
|
| 554 |
+
|
| 555 |
+
### Training Logs
|
| 556 |
+
| Epoch | Step | Training Loss |
|
| 557 |
+
|:------:|:----:|:-------------:|
|
| 558 |
+
| 0.4941 | 500 | 0.8956 |
|
| 559 |
+
| 0.9881 | 1000 | 0.6364 |
|
| 560 |
+
| 1.4822 | 1500 | 0.5192 |
|
| 561 |
+
| 1.9763 | 2000 | 0.4969 |
|
| 562 |
+
| 2.4704 | 2500 | 0.4261 |
|
| 563 |
+
| 2.9644 | 3000 | 0.4208 |
|
| 564 |
+
| 3.4585 | 3500 | 0.3828 |
|
| 565 |
+
| 3.9526 | 4000 | 0.3874 |
|
| 566 |
+
|
| 567 |
+
|
| 568 |
+
### Framework Versions
|
| 569 |
+
- Python: 3.11.12
|
| 570 |
+
- Sentence Transformers: 3.4.1
|
| 571 |
+
- Transformers: 4.51.3
|
| 572 |
+
- PyTorch: 2.6.0+cu124
|
| 573 |
+
- Accelerate: 1.6.0
|
| 574 |
+
- Datasets: 3.5.1
|
| 575 |
+
- Tokenizers: 0.21.1
|
| 576 |
+
|
| 577 |
+
## Citation
|
| 578 |
+
|
| 579 |
+
### BibTeX
|
| 580 |
+
|
| 581 |
+
#### Sentence Transformers
|
| 582 |
+
```bibtex
|
| 583 |
+
@inproceedings{reimers-2019-sentence-bert,
|
| 584 |
+
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
|
| 585 |
+
author = "Reimers, Nils and Gurevych, Iryna",
|
| 586 |
+
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
|
| 587 |
+
month = "11",
|
| 588 |
+
year = "2019",
|
| 589 |
+
publisher = "Association for Computational Linguistics",
|
| 590 |
+
url = "https://arxiv.org/abs/1908.10084",
|
| 591 |
+
}
|
| 592 |
+
```
|
| 593 |
+
|
| 594 |
+
#### MultipleNegativesRankingLoss
|
| 595 |
+
```bibtex
|
| 596 |
+
@misc{henderson2017efficient,
|
| 597 |
+
title={Efficient Natural Language Response Suggestion for Smart Reply},
|
| 598 |
+
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},
|
| 599 |
+
year={2017},
|
| 600 |
+
eprint={1705.00652},
|
| 601 |
+
archivePrefix={arXiv},
|
| 602 |
+
primaryClass={cs.CL}
|
| 603 |
+
}
|
| 604 |
+
```
|
| 605 |
+
|
| 606 |
+
<!--
|
| 607 |
+
## Glossary
|
| 608 |
+
|
| 609 |
+
*Clearly define terms in order to be accessible across audiences.*
|
| 610 |
+
-->
|
| 611 |
+
|
| 612 |
+
<!--
|
| 613 |
+
## Model Card Authors
|
| 614 |
+
|
| 615 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
| 616 |
+
-->
|
| 617 |
+
|
| 618 |
+
<!--
|
| 619 |
+
## Model Card Contact
|
| 620 |
+
|
| 621 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
| 622 |
+
-->
|
config.json
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"BertModel"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"classifier_dropout": null,
|
| 7 |
+
"gradient_checkpointing": false,
|
| 8 |
+
"hidden_act": "gelu",
|
| 9 |
+
"hidden_dropout_prob": 0.1,
|
| 10 |
+
"hidden_size": 384,
|
| 11 |
+
"initializer_range": 0.02,
|
| 12 |
+
"intermediate_size": 1536,
|
| 13 |
+
"layer_norm_eps": 1e-12,
|
| 14 |
+
"max_position_embeddings": 512,
|
| 15 |
+
"model_type": "bert",
|
| 16 |
+
"num_attention_heads": 12,
|
| 17 |
+
"num_hidden_layers": 6,
|
| 18 |
+
"pad_token_id": 0,
|
| 19 |
+
"position_embedding_type": "absolute",
|
| 20 |
+
"torch_dtype": "float32",
|
| 21 |
+
"transformers_version": "4.51.3",
|
| 22 |
+
"type_vocab_size": 2,
|
| 23 |
+
"use_cache": true,
|
| 24 |
+
"vocab_size": 30522
|
| 25 |
+
}
|
config_sentence_transformers.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"__version__": {
|
| 3 |
+
"sentence_transformers": "3.4.1",
|
| 4 |
+
"transformers": "4.51.3",
|
| 5 |
+
"pytorch": "2.6.0+cu124"
|
| 6 |
+
},
|
| 7 |
+
"prompts": {},
|
| 8 |
+
"default_prompt_name": null,
|
| 9 |
+
"similarity_fn_name": "cosine"
|
| 10 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:985c046eb9305d68ae80b0da47a81fc551d59bb2fcf54e7e051cc1e24defb3a0
|
| 3 |
+
size 90864192
|
modules.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"idx": 0,
|
| 4 |
+
"name": "0",
|
| 5 |
+
"path": "",
|
| 6 |
+
"type": "sentence_transformers.models.Transformer"
|
| 7 |
+
},
|
| 8 |
+
{
|
| 9 |
+
"idx": 1,
|
| 10 |
+
"name": "1",
|
| 11 |
+
"path": "1_Pooling",
|
| 12 |
+
"type": "sentence_transformers.models.Pooling"
|
| 13 |
+
},
|
| 14 |
+
{
|
| 15 |
+
"idx": 2,
|
| 16 |
+
"name": "2",
|
| 17 |
+
"path": "2_Normalize",
|
| 18 |
+
"type": "sentence_transformers.models.Normalize"
|
| 19 |
+
}
|
| 20 |
+
]
|
sentence_bert_config.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"max_seq_length": 256,
|
| 3 |
+
"do_lower_case": false
|
| 4 |
+
}
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cls_token": {
|
| 3 |
+
"content": "[CLS]",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"mask_token": {
|
| 10 |
+
"content": "[MASK]",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"pad_token": {
|
| 17 |
+
"content": "[PAD]",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"sep_token": {
|
| 24 |
+
"content": "[SEP]",
|
| 25 |
+
"lstrip": false,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"unk_token": {
|
| 31 |
+
"content": "[UNK]",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
}
|
| 37 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"added_tokens_decoder": {
|
| 3 |
+
"0": {
|
| 4 |
+
"content": "[PAD]",
|
| 5 |
+
"lstrip": false,
|
| 6 |
+
"normalized": false,
|
| 7 |
+
"rstrip": false,
|
| 8 |
+
"single_word": false,
|
| 9 |
+
"special": true
|
| 10 |
+
},
|
| 11 |
+
"100": {
|
| 12 |
+
"content": "[UNK]",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"101": {
|
| 20 |
+
"content": "[CLS]",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"102": {
|
| 28 |
+
"content": "[SEP]",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"103": {
|
| 36 |
+
"content": "[MASK]",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"clean_up_tokenization_spaces": false,
|
| 45 |
+
"cls_token": "[CLS]",
|
| 46 |
+
"do_basic_tokenize": true,
|
| 47 |
+
"do_lower_case": true,
|
| 48 |
+
"extra_special_tokens": {},
|
| 49 |
+
"mask_token": "[MASK]",
|
| 50 |
+
"max_length": 128,
|
| 51 |
+
"model_max_length": 256,
|
| 52 |
+
"never_split": null,
|
| 53 |
+
"pad_to_multiple_of": null,
|
| 54 |
+
"pad_token": "[PAD]",
|
| 55 |
+
"pad_token_type_id": 0,
|
| 56 |
+
"padding_side": "right",
|
| 57 |
+
"sep_token": "[SEP]",
|
| 58 |
+
"stride": 0,
|
| 59 |
+
"strip_accents": null,
|
| 60 |
+
"tokenize_chinese_chars": true,
|
| 61 |
+
"tokenizer_class": "BertTokenizer",
|
| 62 |
+
"truncation_side": "right",
|
| 63 |
+
"truncation_strategy": "longest_first",
|
| 64 |
+
"unk_token": "[UNK]"
|
| 65 |
+
}
|
vocab.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|