Upload embedding model
Browse files- 1_Pooling/config.json +10 -0
- README.md +593 -0
- config.json +30 -0
- config_sentence_transformers.json +14 -0
- model.safetensors +3 -0
- modules.json +20 -0
- sentence_bert_config.json +4 -0
- tokenizer.json +0 -0
- tokenizer_config.json +16 -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,593 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
tags:
|
| 5 |
+
- sentence-transformers
|
| 6 |
+
- sentence-similarity
|
| 7 |
+
- feature-extraction
|
| 8 |
+
- dense
|
| 9 |
+
- generated_from_trainer
|
| 10 |
+
- dataset_size:59315
|
| 11 |
+
- loss:MultipleNegativesRankingLoss
|
| 12 |
+
base_model: sentence-transformers/all-MiniLM-L6-v2
|
| 13 |
+
widget:
|
| 14 |
+
- source_sentence: Johnny Depp plays policeman Ichabod Crane in which 1999 film?
|
| 15 |
+
sentences:
|
| 16 |
+
- 'Mythology in France epics and fairy tales as part of deeply embedded spiritual
|
| 17 |
+
allegories and mythological archetypes: Mythology in France The mythologies in
|
| 18 |
+
present-day France encompass the mythology of the Gauls, Franks, Normans, Bretons,
|
| 19 |
+
and other peoples living in France, those ancient stories about divine or heroic
|
| 20 |
+
beings that these particular cultures believed to be true and that often use supernatural
|
| 21 |
+
events or characters to explain the nature of the universe and humanity. French
|
| 22 |
+
mythology is listed for each culture. Bretons are a subset of the celtics that
|
| 23 |
+
adopted Christianity. Celtic cosmology predominates their mythology: Gauls were
|
| 24 |
+
another subset of Celtic people. Celtic'
|
| 25 |
+
- Johnny Depp in a snuff film in exchange for money for his family. Depp was a fan
|
| 26 |
+
and friend of writer Hunter S. Thompson, and played his alter ego Raoul Duke in
|
| 27 |
+
"Fear and Loathing in Las Vegas" (1998), Terry Gilliam's film adaptation of Thompson's
|
| 28 |
+
pseudobiographical novel of the same name. Depp's next venture with Burton was
|
| 29 |
+
the period film "Sleepy Hollow" (1999), in which he played Ichabod Crane opposite
|
| 30 |
+
Christina Ricci and Christopher Walken. For his performance, Depp took inspiration
|
| 31 |
+
from Angela Lansbury, Roddy McDowall and Basil Rathbone. He stated that he "always
|
| 32 |
+
thought of Ichabod as a very delicate, fragile
|
| 33 |
+
- Ichabod Crane Kinderhook town school district (Ichabod Crane Central School District)
|
| 34 |
+
is also named for the Irving character. It is claimed by many in Tarrytown that
|
| 35 |
+
Samuel Youngs is the original from whom Irving drew his character of Ichabod Crane".
|
| 36 |
+
Author Gary Denis asserts that while the character of Ichabod Crane is loosely
|
| 37 |
+
based on Kinderhook Schoolmaster, Jesse Merwin, it may possibly include elements
|
| 38 |
+
from Samuel Youngs' life. Irving's characters drive the story and are most memorable
|
| 39 |
+
because of his detail in describing each. He says of Ichabod Crane (the main character),
|
| 40 |
+
'He was tall, but exceedingly lank, with narrow shoulders, long
|
| 41 |
+
- source_sentence: What is the name of Liam Gallagher's new band, whose first album
|
| 42 |
+
is 'Different Gear, Still Speeding'?
|
| 43 |
+
sentences:
|
| 44 |
+
- Different Gear, Still Speeding Different Gear, Still Speeding Different Gear,
|
| 45 |
+
Still Speeding is the debut studio album by English rock band Beady Eye, released
|
| 46 |
+
on 28 February 2011. It debuted at number three in the UK Albums Chart selling
|
| 47 |
+
66,817 in the first week. As of August 2012, the album has sold 174,487 copies
|
| 48 |
+
in the UK. On "Different Gear, Still Speeding", all members contributed to the
|
| 49 |
+
instrumentation, much like the later albums of Oasis. Reviews of the album have
|
| 50 |
+
been generally mixed-to-favourable. According to review aggregator site Metacritic,
|
| 51 |
+
the album has an average score of 65%. Reviewing for "Rolling Stone", Stacey Anderson,
|
| 52 |
+
who
|
| 53 |
+
- Mickey Mouse universe was originally going to be named Mortimer. Lillian Disney,
|
| 54 |
+
Walt's wife, suggested the name Mickey instead. The first Mortimer was created
|
| 55 |
+
by Walt Disney and Floyd Gottfredson for the comics. He was Minnie Mouse's ranch-owning
|
| 56 |
+
cattleman uncle. He first appeared in the serial "Mickey Mouse in Death Valley"
|
| 57 |
+
(1930). After that, he appeared or was referenced in many other Mickey Mouse comic
|
| 58 |
+
strip adventures in the 1930s. He has occasionally appeared in more modern comics.
|
| 59 |
+
In the 1936 cartoon short "Mickey's Rival", the second Mortimer was introduced
|
| 60 |
+
as Mickey's competitor for Minnie's affections. In the comics, this Mortimer was
|
| 61 |
+
briefly
|
| 62 |
+
- Different Gear, Still Speeding gave the album 2.5 out of 5 stars, said "On Different
|
| 63 |
+
Gear, the band attempts stripped down, Stones-y rock but ends up with 'Be Here
|
| 64 |
+
Now'-style guitar bluster and Liam's blithely boilerplate lyrics". Drowned in
|
| 65 |
+
Sound awarded the album 4/10 saying that "By and large it radiates the stolid
|
| 66 |
+
competence of a band on auto-pilot, with a few flashes of likeable enthusiasm."
|
| 67 |
+
The "Independent on Sunday" gave it 2/5 stars. Simon Goddard, reviewing for "Q",
|
| 68 |
+
gave the album four-out-of-five stars and described it as "the strongest record
|
| 69 |
+
Liam's made" since "(What's the Story) Morning Glory?", while Garry Mulholland,
|
| 70 |
+
in his
|
| 71 |
+
- source_sentence: The bands R.E.M. and the B52s both come from which town in Georgia?
|
| 72 |
+
sentences:
|
| 73 |
+
- 'E and M signaling by the resistance of the wire, but will normally be less than
|
| 74 |
+
100m for adequate noise immunity. The group of E&M signaling includes the following
|
| 75 |
+
variations: E&M defines eight wires: "4-wire E&M" uses a 4-wire (2-pair) transmission
|
| 76 |
+
path for the voice signal. "2-wire E&M" uses a single pair for both transmit and
|
| 77 |
+
receive voice signal. This is much inferior to 4-wire E&M as the 2-wire interface
|
| 78 |
+
uses hybrid transformers which reduce signal quality and can introduce echo. The
|
| 79 |
+
mechanisms described so far only allow circuit seizure – on-hook and off-hook
|
| 80 |
+
– to be signaled. In order to allow dialing over'
|
| 81 |
+
- Culture of Georgia (U.S. state) a fertile field for alternative rock bands since
|
| 82 |
+
the late 1970s. Notable bands from Athens include R.E.M., The B-52's, Widespread
|
| 83 |
+
Panic, Drive-By Truckers, as well as bands from the Elephant 6 Recording Company
|
| 84 |
+
most notably Neutral Milk Hotel. Rhythm and Blues is another important musical
|
| 85 |
+
genre in Georgia. Ray Charles was one of popular music's most influential performers,
|
| 86 |
+
fusing R&B, jazz, and country into many popular songs. Augusta native James Brown
|
| 87 |
+
and Macon native Little Richard, two important figures in R&B history, started
|
| 88 |
+
performing in Georgia clubs on the chitlin' circuit, fused gospel with blues and
|
| 89 |
+
boogie-woogie to lay the
|
| 90 |
+
- 'I Just Shot John Lennon I Just Shot John Lennon "I Just Shot John Lennon" is
|
| 91 |
+
a song from The Cranberries'' album "To the Faithful Departed". It is a narrative
|
| 92 |
+
of the events of the night of December 8, 1980, the night that musician John Lennon
|
| 93 |
+
was murdered by Mark David Chapman in front of The Dakota in New York City. It
|
| 94 |
+
is one of many tributes to Lennon, and also one of many other songs to recall
|
| 95 |
+
the events of the night. After the narrative, there is commentary: "What a sad,
|
| 96 |
+
and sorry and sickening sight". The title of the song comes from the'
|
| 97 |
+
- source_sentence: Who wrote Three Men In A Boat?
|
| 98 |
+
sentences:
|
| 99 |
+
- Three Men in a Boat Three Men in a Boat Three Men in a Boat (To Say Nothing of
|
| 100 |
+
the Dog), published in 1889, is a humorous account by English writer Jerome K.
|
| 101 |
+
Jerome of a two-week boating holiday on the Thames from Kingston upon Thames to
|
| 102 |
+
Oxford and back to Kingston. The book was initially intended to be a serious travel
|
| 103 |
+
guide, with accounts of local history along the route, but the humorous elements
|
| 104 |
+
took over to the point where the serious and somewhat sentimental passages seem
|
| 105 |
+
a distraction to the comic novel. One of the most praised things about "Three
|
| 106 |
+
Men in a
|
| 107 |
+
- 'Darwin, Northern Territory line. Darwin lies in the Northern Territory, on the
|
| 108 |
+
Timor Sea. The city proper occupies a low bluff overlooking Darwin Harbour, flanked
|
| 109 |
+
by Frances Bay to the east and Cullen Bay to the west. The remainder of the city
|
| 110 |
+
is flat and low-lying, and coastal areas are home to recreational reserves, extensive
|
| 111 |
+
beaches, and excellent fishing. Darwin is closer to the capitals of five other
|
| 112 |
+
countries than to the capital of Australia: Darwin is away from Canberra. Dili
|
| 113 |
+
(East Timor) is , Port Moresby (Papua New Guinea) is , Jakarta (Indonesia) is
|
| 114 |
+
, Bandar Seri Begawan (Brunei) is , and'
|
| 115 |
+
- 'Three Men in a Boat 1891, "Three Women in One Boat: A River Sketch" by Constance
|
| 116 |
+
MacEwen was published. This book relates the journey of three young university
|
| 117 |
+
women who set out to emulate the river trip in "Three Men in a Boat" in an effort
|
| 118 |
+
to raise the spirits of one of them, who is about to be expelled from university.
|
| 119 |
+
To take the place of Montmorency, they bring a cat called Tintoretto. Three Men
|
| 120 |
+
in a Boat is referenced in the 1956 parody novel on mountaineering, "The Ascent
|
| 121 |
+
of Rum Doodle", where the head porter Bing is said to spend "much of his'
|
| 122 |
+
- source_sentence: In the RAF, what is the rank immediately above Squadron Leader?
|
| 123 |
+
sentences:
|
| 124 |
+
- Tybalt Tybalt Tybalt is the main antagonist in William Shakespeare's play "Romeo
|
| 125 |
+
and Juliet". He is the son of Lady Capulet's brother, Juliet's short-tempered
|
| 126 |
+
first cousin, and Romeo's rival. Tybalt shares the same name as the character
|
| 127 |
+
Tibert/Tybalt the "Prince of Cats" in "Reynard the Fox", a point of mockery in
|
| 128 |
+
the play. Mercutio repeatedly calls Tybalt "Prince of Cats" (perhaps referring
|
| 129 |
+
not only to Reynard but to the Italian word cazzo as well). Luigi da Porto adapted
|
| 130 |
+
the story as "Giulietta e Romeo" and included it in his "Historia novellamente
|
| 131 |
+
ritrovata di due Nobili Amanti" published in 1530. Da Porto
|
| 132 |
+
- Squadron leader RAF used major as the equivalent rank to squadron leader. Royal
|
| 133 |
+
Naval Air Service lieutenant-commanders and Royal Flying Corps majors on 31 March
|
| 134 |
+
1918 became RAF majors on 1 April 1918. On 31 August 1919, the RAF rank of major
|
| 135 |
+
was superseded by squadron leader which has remained in continuous usage ever
|
| 136 |
+
since. Promotion to squadron leader is strictly on merit, and requires the individual
|
| 137 |
+
to be appointed to a Career Commission, which will see them remain in the RAF
|
| 138 |
+
until retirement or voluntary resignation. Before the Second World War, a squadron
|
| 139 |
+
leader commanded a squadron of aircraft. Today, however,
|
| 140 |
+
- Squadron leader Squadron leader Squadron leader (Sqn Ldr in the RAF ; SQNLDR in
|
| 141 |
+
the RAAF and RNZAF; formerly sometimes S/L in all services) is a commissioned
|
| 142 |
+
rank in the Royal Air Force and the air forces of many countries which have historical
|
| 143 |
+
British influence. It is also sometimes used as the English translation of an
|
| 144 |
+
equivalent rank in countries which have a non-English air force-specific rank
|
| 145 |
+
structure. An air force squadron leader ranks above flight lieutenant and immediately
|
| 146 |
+
below wing commander and it is the most junior of the senior officer ranks. The
|
| 147 |
+
air force rank of squadron leader has a
|
| 148 |
+
datasets:
|
| 149 |
+
- sentence-transformers/trivia-qa-triplet
|
| 150 |
+
pipeline_tag: sentence-similarity
|
| 151 |
+
library_name: sentence-transformers
|
| 152 |
+
metrics:
|
| 153 |
+
- cosine_accuracy
|
| 154 |
+
model-index:
|
| 155 |
+
- name: SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
|
| 156 |
+
results:
|
| 157 |
+
- task:
|
| 158 |
+
type: triplet
|
| 159 |
+
name: Triplet
|
| 160 |
+
dataset:
|
| 161 |
+
name: trivia qa eval
|
| 162 |
+
type: trivia_qa_eval
|
| 163 |
+
metrics:
|
| 164 |
+
- type: cosine_accuracy
|
| 165 |
+
value: 0.8339999914169312
|
| 166 |
+
name: Cosine Accuracy
|
| 167 |
+
---
|
| 168 |
+
|
| 169 |
+
# SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
|
| 170 |
+
|
| 171 |
+
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) on the [trivia-qa-triplet](https://huggingface.co/datasets/sentence-transformers/trivia-qa-triplet) dataset. 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.
|
| 172 |
+
|
| 173 |
+
## Model Details
|
| 174 |
+
|
| 175 |
+
### Model Description
|
| 176 |
+
- **Model Type:** Sentence Transformer
|
| 177 |
+
- **Base model:** [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) <!-- at revision c9745ed1d9f207416be6d2e6f8de32d1f16199bf -->
|
| 178 |
+
- **Maximum Sequence Length:** 256 tokens
|
| 179 |
+
- **Output Dimensionality:** 384 dimensions
|
| 180 |
+
- **Similarity Function:** Cosine Similarity
|
| 181 |
+
- **Training Dataset:**
|
| 182 |
+
- [trivia-qa-triplet](https://huggingface.co/datasets/sentence-transformers/trivia-qa-triplet)
|
| 183 |
+
- **Language:** en
|
| 184 |
+
<!-- - **License:** Unknown -->
|
| 185 |
+
|
| 186 |
+
### Model Sources
|
| 187 |
+
|
| 188 |
+
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
| 189 |
+
- **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
|
| 190 |
+
- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
| 191 |
+
|
| 192 |
+
### Full Model Architecture
|
| 193 |
+
|
| 194 |
+
```
|
| 195 |
+
SentenceTransformer(
|
| 196 |
+
(0): Transformer({'max_seq_length': 256, 'do_lower_case': False, 'architecture': 'BertModel'})
|
| 197 |
+
(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})
|
| 198 |
+
(2): Normalize()
|
| 199 |
+
)
|
| 200 |
+
```
|
| 201 |
+
|
| 202 |
+
## Usage
|
| 203 |
+
|
| 204 |
+
### Direct Usage (Sentence Transformers)
|
| 205 |
+
|
| 206 |
+
First install the Sentence Transformers library:
|
| 207 |
+
|
| 208 |
+
```bash
|
| 209 |
+
pip install -U sentence-transformers
|
| 210 |
+
```
|
| 211 |
+
|
| 212 |
+
Then you can load this model and run inference.
|
| 213 |
+
```python
|
| 214 |
+
from sentence_transformers import SentenceTransformer
|
| 215 |
+
|
| 216 |
+
# Download from the 🤗 Hub
|
| 217 |
+
model = SentenceTransformer("sentence_transformers_model_id")
|
| 218 |
+
# Run inference
|
| 219 |
+
sentences = [
|
| 220 |
+
'In the RAF, what is the rank immediately above Squadron Leader?',
|
| 221 |
+
'Squadron leader Squadron leader Squadron leader (Sqn Ldr in the RAF ; SQNLDR in the RAAF and RNZAF; formerly sometimes S/L in all services) is a commissioned rank in the Royal Air Force and the air forces of many countries which have historical British influence. It is also sometimes used as the English translation of an equivalent rank in countries which have a non-English air force-specific rank structure. An air force squadron leader ranks above flight lieutenant and immediately below wing commander and it is the most junior of the senior officer ranks. The air force rank of squadron leader has a',
|
| 222 |
+
'Squadron leader RAF used major as the equivalent rank to squadron leader. Royal Naval Air Service lieutenant-commanders and Royal Flying Corps majors on 31 March 1918 became RAF majors on 1 April 1918. On 31 August 1919, the RAF rank of major was superseded by squadron leader which has remained in continuous usage ever since. Promotion to squadron leader is strictly on merit, and requires the individual to be appointed to a Career Commission, which will see them remain in the RAF until retirement or voluntary resignation. Before the Second World War, a squadron leader commanded a squadron of aircraft. Today, however,',
|
| 223 |
+
]
|
| 224 |
+
embeddings = model.encode(sentences)
|
| 225 |
+
print(embeddings.shape)
|
| 226 |
+
# [3, 384]
|
| 227 |
+
|
| 228 |
+
# Get the similarity scores for the embeddings
|
| 229 |
+
similarities = model.similarity(embeddings, embeddings)
|
| 230 |
+
print(similarities)
|
| 231 |
+
# tensor([[1.0000, 0.5605, 0.4394],
|
| 232 |
+
# [0.5605, 1.0000, 0.5768],
|
| 233 |
+
# [0.4394, 0.5768, 1.0000]])
|
| 234 |
+
```
|
| 235 |
+
|
| 236 |
+
<!--
|
| 237 |
+
### Direct Usage (Transformers)
|
| 238 |
+
|
| 239 |
+
<details><summary>Click to see the direct usage in Transformers</summary>
|
| 240 |
+
|
| 241 |
+
</details>
|
| 242 |
+
-->
|
| 243 |
+
|
| 244 |
+
<!--
|
| 245 |
+
### Downstream Usage (Sentence Transformers)
|
| 246 |
+
|
| 247 |
+
You can finetune this model on your own dataset.
|
| 248 |
+
|
| 249 |
+
<details><summary>Click to expand</summary>
|
| 250 |
+
|
| 251 |
+
</details>
|
| 252 |
+
-->
|
| 253 |
+
|
| 254 |
+
<!--
|
| 255 |
+
### Out-of-Scope Use
|
| 256 |
+
|
| 257 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 258 |
+
-->
|
| 259 |
+
|
| 260 |
+
## Evaluation
|
| 261 |
+
|
| 262 |
+
### Metrics
|
| 263 |
+
|
| 264 |
+
#### Triplet
|
| 265 |
+
|
| 266 |
+
* Dataset: `trivia_qa_eval`
|
| 267 |
+
* Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.TripletEvaluator)
|
| 268 |
+
|
| 269 |
+
| Metric | Value |
|
| 270 |
+
|:--------------------|:----------|
|
| 271 |
+
| **cosine_accuracy** | **0.834** |
|
| 272 |
+
|
| 273 |
+
<!--
|
| 274 |
+
## Bias, Risks and Limitations
|
| 275 |
+
|
| 276 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| 277 |
+
-->
|
| 278 |
+
|
| 279 |
+
<!--
|
| 280 |
+
### Recommendations
|
| 281 |
+
|
| 282 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| 283 |
+
-->
|
| 284 |
+
|
| 285 |
+
## Training Details
|
| 286 |
+
|
| 287 |
+
### Training Dataset
|
| 288 |
+
|
| 289 |
+
#### trivia-qa-triplet
|
| 290 |
+
|
| 291 |
+
* Dataset: [trivia-qa-triplet](https://huggingface.co/datasets/sentence-transformers/trivia-qa-triplet) at [bfe9460](https://huggingface.co/datasets/sentence-transformers/trivia-qa-triplet/tree/bfe94607eb149a89fe8107e8c8d187977e587a7d)
|
| 292 |
+
* Size: 59,315 training samples
|
| 293 |
+
* Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
|
| 294 |
+
* Approximate statistics based on the first 1000 samples:
|
| 295 |
+
| | anchor | positive | negative |
|
| 296 |
+
|:--------|:----------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------|
|
| 297 |
+
| type | string | string | string |
|
| 298 |
+
| details | <ul><li>min: 8 tokens</li><li>mean: 20.08 tokens</li><li>max: 71 tokens</li></ul> | <ul><li>min: 114 tokens</li><li>mean: 139.64 tokens</li><li>max: 226 tokens</li></ul> | <ul><li>min: 113 tokens</li><li>mean: 138.81 tokens</li><li>max: 256 tokens</li></ul> |
|
| 299 |
+
* Samples:
|
| 300 |
+
| anchor | positive | negative |
|
| 301 |
+
|:-----------------------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
| 302 |
+
| <code>"Sirius otherwise known as the ""Dog Star"" can be found in which constellation?"</code> | <code>Sirius in fiction below)—and even if such were eventually discovered, with an estimated age of 230 million years the system is too young to have fostered the development of life or a complex biosphere. Sirius AB is the alpha star of the constellation Canis Major (the great dog, sometimes styled as Orion's hunting dog), whence its cognomen "the dog star". The most commonly used proper name of this star comes through the Latin "Sirius", from the Greek Σείριος ("Seirios", "glowing" or "scorcher"). The ancient Greeks observed that the appearance of Sirius heralded the hot and dry "dog days" of summer, and feared that</code> | <code>Dog days "Seírios", "Scorcher"), Sothis (, "Sôthis", a transcription of Egyptian "Spdt"), and the Dog Star (, "Kúōn"). The last name reflects the way Sirius follows the constellation Orion into the night sky. Sirius is by far the brightest proper star in the night sky, which caused ancient astronomers to take note of it around the world. In Egypt, its return to the night sky became known as a precursor to the annual flooding of the Nile and was worshipped as the goddess Sopdet. In Greece, it became known as the precursor of the unpleasantly hot phase of the summer. Greek poets</code> |
|
| 303 |
+
| <code>Which Scottish town is the administrative centre of the Highland Region?</code> | <code>Highland Scottish All operations are now part of the Stagecoach Group. Highland Scottish Highland Scottish Omnibuses Ltd was formed as a bus operating subsidiary of the Scottish Transport Group in June 1985 from Highland Omnibuses Ltd, and operated until October 1995 when the company was split into two - Highland Bus & Coach and Highland Country Buses. The companies have since remerged and operate today as Highland Country Buses. From its head office in Seafield Road, Inverness, Highland Scottish operated over the massive geographical, but sparsely populated, area of the Highland region of north west Scotland. Highland Scottish was the largest operator</code> | <code>Highland Scottish in the centre of its 'Highland Country' logo. In January 1996 Highland Country Buses was bought by National Express for £1.8m. Highland Bus & Coach, being the smaller of the two operators, continued to operate with the image its predecessor adopted. The two companies continued to exist under separate ownership until August 1998 when Rapson's bought Highland Country Buses back from National Express for £4m - £2.2m more than Rapson's originally sold the company for. Highland Country Buses is now a wholly owned subsidiary of Rapson's Coaches, and covers the operating area that Highland Scottish had on privatisation. The company</code> |
|
| 304 |
+
| <code>Which brand of coffee is named after a hotel in Nashville, Tennessee?</code> | <code>History of Nashville, Tennessee proprietor thereof, had served a special blend of coffee at the hotel's restaurant, and after drinking a cup of this coffee, Roosevelt proclaimed it "good to the last drop!" Cheek subsequently sold the blend to General Foods and to this day, Maxwell House coffee is enjoyed by millions. In 1913 Nashville was the last of several major cities in Tennessee to adopt a commission form of government, with all members of a small commission elected at-large. Compared to single-member districts, this change resulted in further limiting the political power of any African Americans who were able to vote, as their</code> | <code>Renaissance Nashville Hotel Renaissance Nashville Hotel The Renaissance Nashville Hotel is a hotel in Nashville, Tennessee. The building is 385 feet high with 31 floors. The hotel is physically connected to the Nashville Convention Center and is its anchor hotel. The hotel contains 649 rooms, 24 suites, 25 meeting rooms with 31,000 sq ft of meeting space, and 2 concierge levels including a Starbucks coffee shop, 2 lounges, and a full service restaurant. One of the lounges is located in an enclosed bridge walkway, spanning above Commerce Street, which connects the hotel to a parking garage across the street. This walkway was severely</code> |
|
| 305 |
+
* Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
|
| 306 |
+
```json
|
| 307 |
+
{
|
| 308 |
+
"scale": 20.0,
|
| 309 |
+
"similarity_fct": "cos_sim",
|
| 310 |
+
"gather_across_devices": false
|
| 311 |
+
}
|
| 312 |
+
```
|
| 313 |
+
|
| 314 |
+
### Evaluation Dataset
|
| 315 |
+
|
| 316 |
+
#### trivia-qa-triplet
|
| 317 |
+
|
| 318 |
+
* Dataset: [trivia-qa-triplet](https://huggingface.co/datasets/sentence-transformers/trivia-qa-triplet) at [bfe9460](https://huggingface.co/datasets/sentence-transformers/trivia-qa-triplet/tree/bfe94607eb149a89fe8107e8c8d187977e587a7d)
|
| 319 |
+
* Size: 1,000 evaluation samples
|
| 320 |
+
* Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
|
| 321 |
+
* Approximate statistics based on the first 1000 samples:
|
| 322 |
+
| | anchor | positive | negative |
|
| 323 |
+
|:--------|:-----------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------|
|
| 324 |
+
| type | string | string | string |
|
| 325 |
+
| details | <ul><li>min: 8 tokens</li><li>mean: 20.18 tokens</li><li>max: 103 tokens</li></ul> | <ul><li>min: 111 tokens</li><li>mean: 139.08 tokens</li><li>max: 225 tokens</li></ul> | <ul><li>min: 113 tokens</li><li>mean: 138.94 tokens</li><li>max: 256 tokens</li></ul> |
|
| 326 |
+
* Samples:
|
| 327 |
+
| anchor | positive | negative |
|
| 328 |
+
|:---------------------------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
| 329 |
+
| <code>What African country is projected to pass the United States in population by the year 2055?</code> | <code>African immigration to the United States entering the United States. It is estimated that the current population of African immigrants to the United States is about 2.1 million. According to the Migration Policy Institute, as of 2009 two-thirds of the African immigrants were from either East or West Africa. Countries with the most immigrants to the U.S. are Nigeria, Egypt, Ethiopia, Ghana, South Africa, Somalia, Eritrea, and Kenya. Seventy five percent (75%) of the African immigrants to the U.S. come from 12 of the 55 countries, namely Nigeria, Egypt, Ghana, Ethiopia, South Africa, Kenya, Liberia, Somalia, Morocco, Cape Verde, Sierra Leone and Sudan (including what is</code> | <code>African immigration to the United States African countries due to many skilled hard-working Africans leaving Africa to seek their economic fortunes in the U.S. mainly and elsewhere. One major factor that contributes to migration from Africa to the United States is labor opportunities. It has been relatively easier for African immigrants for Africans with advanced education to leave and enter international labor markets. In addition, many Africans move to the United States for advanced training. For example, doctors from different African nations would move to the U.S. in order to gain more economic opportunities compared to their home country. However, as more Africans emigrate to the</code> |
|
| 330 |
+
| <code>Which is the largest species of the turtle family?</code> | <code>Leatherback sea turtle Leatherback sea turtle The leatherback sea turtle ("Dermochelys coriacea"), sometimes called the lute turtle or leathery turtle or simply the luth, is the largest of all living turtles and is the fourth-heaviest modern reptile behind three crocodilians. It is the only living species in the genus Dermochelys and family Dermochelyidae. It can easily be differentiated from other modern sea turtles by its lack of a bony shell, hence the name. Instead, its carapace is covered by skin and oily flesh. "Dermochelys" is the only extant genus of the family Dermochelyidae. "Dermochelys coriacea" is the only species in genus "Dermochelys". The</code> | <code>Okavango mud turtle Okavango mud turtle The Okavango mud turtle (Okavango terrapin) ("Pelusios bechuanicus") is a species of turtle in the family Pelomedusidae endemic to Africa. It is found in Angola, Botswana, the Democratic Republic of the Congo, Namibia (Caprivi), Zambia, and Zimbabwe. Found in central Africa, central Angola, northeastern Namibia, northern Botswana, Zimbabwe, and Zambia The Okavango mud turtle is largest species of the genus "Pelusios". The carapace is oval and elongated, with a pronounced dome, and is evenly rounded at the edges which allows the turtle to appear as a smooth rock. The carapace is very dark, often almost black, and</code> |
|
| 331 |
+
| <code>How many gallons of beer are in an English barrel?</code> | <code>Barrel gallon for liquids (the corn gallon of 268.8 cubic inches for solids). In Britain, the wine gallon was replaced by the imperial gallon. The tierce later became the petrol barrel. The tun was originally 256 gallons, which explains from where the quarter, 8 bushels or 64 (wine) gallons, comes. Although it is common to refer to draught beer containers of any size as barrels, in the UK this is strictly correct only if the container holds 36 imperial gallons. The terms "keg" and "cask" refer to containers of any size, the distinction being that kegs are used for beers intended</code> | <code>Beer Barrel Polecats Percy Pomeroy (Eddie Laughton), and work together to flee the prison. They are ultimately captured, and sent to solitary confinement. After nearly half a century later, the graying trio are finally released as senior citizens, in which Curly quips upon leaving "You know what I'm-a gonna do? I'm gonna get myself a tall, big, beautiful bottle of beer!" Moe and Larry become irate and throw Curly back into the jail, leaving him there. The title "Beer Barrel Polecats" is a pun of the song "Beer Barrel Polka". The idea of producing and selling their own beer during Prohibition was borrowed</code> |
|
| 332 |
+
* Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
|
| 333 |
+
```json
|
| 334 |
+
{
|
| 335 |
+
"scale": 20.0,
|
| 336 |
+
"similarity_fct": "cos_sim",
|
| 337 |
+
"gather_across_devices": false
|
| 338 |
+
}
|
| 339 |
+
```
|
| 340 |
+
|
| 341 |
+
### Training Hyperparameters
|
| 342 |
+
#### Non-Default Hyperparameters
|
| 343 |
+
|
| 344 |
+
- `eval_strategy`: steps
|
| 345 |
+
- `per_device_train_batch_size`: 64
|
| 346 |
+
- `per_device_eval_batch_size`: 64
|
| 347 |
+
- `learning_rate`: 2e-05
|
| 348 |
+
- `weight_decay`: 0.01
|
| 349 |
+
- `num_train_epochs`: 4
|
| 350 |
+
- `warmup_ratio`: 0.1
|
| 351 |
+
- `warmup_steps`: 0.1
|
| 352 |
+
- `fp16`: True
|
| 353 |
+
- `load_best_model_at_end`: True
|
| 354 |
+
|
| 355 |
+
#### All Hyperparameters
|
| 356 |
+
<details><summary>Click to expand</summary>
|
| 357 |
+
|
| 358 |
+
- `do_predict`: False
|
| 359 |
+
- `eval_strategy`: steps
|
| 360 |
+
- `prediction_loss_only`: True
|
| 361 |
+
- `per_device_train_batch_size`: 64
|
| 362 |
+
- `per_device_eval_batch_size`: 64
|
| 363 |
+
- `gradient_accumulation_steps`: 1
|
| 364 |
+
- `eval_accumulation_steps`: None
|
| 365 |
+
- `torch_empty_cache_steps`: None
|
| 366 |
+
- `learning_rate`: 2e-05
|
| 367 |
+
- `weight_decay`: 0.01
|
| 368 |
+
- `adam_beta1`: 0.9
|
| 369 |
+
- `adam_beta2`: 0.999
|
| 370 |
+
- `adam_epsilon`: 1e-08
|
| 371 |
+
- `max_grad_norm`: 1.0
|
| 372 |
+
- `num_train_epochs`: 4
|
| 373 |
+
- `max_steps`: -1
|
| 374 |
+
- `lr_scheduler_type`: linear
|
| 375 |
+
- `lr_scheduler_kwargs`: None
|
| 376 |
+
- `warmup_ratio`: 0.1
|
| 377 |
+
- `warmup_steps`: 0.1
|
| 378 |
+
- `log_level`: passive
|
| 379 |
+
- `log_level_replica`: warning
|
| 380 |
+
- `log_on_each_node`: True
|
| 381 |
+
- `logging_nan_inf_filter`: True
|
| 382 |
+
- `enable_jit_checkpoint`: False
|
| 383 |
+
- `save_on_each_node`: False
|
| 384 |
+
- `save_only_model`: False
|
| 385 |
+
- `restore_callback_states_from_checkpoint`: False
|
| 386 |
+
- `use_cpu`: False
|
| 387 |
+
- `seed`: 42
|
| 388 |
+
- `data_seed`: None
|
| 389 |
+
- `bf16`: False
|
| 390 |
+
- `fp16`: True
|
| 391 |
+
- `bf16_full_eval`: False
|
| 392 |
+
- `fp16_full_eval`: False
|
| 393 |
+
- `tf32`: None
|
| 394 |
+
- `local_rank`: -1
|
| 395 |
+
- `ddp_backend`: None
|
| 396 |
+
- `debug`: []
|
| 397 |
+
- `dataloader_drop_last`: False
|
| 398 |
+
- `dataloader_num_workers`: 0
|
| 399 |
+
- `dataloader_prefetch_factor`: None
|
| 400 |
+
- `disable_tqdm`: False
|
| 401 |
+
- `remove_unused_columns`: True
|
| 402 |
+
- `label_names`: None
|
| 403 |
+
- `load_best_model_at_end`: True
|
| 404 |
+
- `ignore_data_skip`: False
|
| 405 |
+
- `fsdp`: []
|
| 406 |
+
- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
|
| 407 |
+
- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
|
| 408 |
+
- `parallelism_config`: None
|
| 409 |
+
- `deepspeed`: None
|
| 410 |
+
- `label_smoothing_factor`: 0.0
|
| 411 |
+
- `optim`: adamw_torch_fused
|
| 412 |
+
- `optim_args`: None
|
| 413 |
+
- `group_by_length`: False
|
| 414 |
+
- `length_column_name`: length
|
| 415 |
+
- `project`: huggingface
|
| 416 |
+
- `trackio_space_id`: trackio
|
| 417 |
+
- `ddp_find_unused_parameters`: None
|
| 418 |
+
- `ddp_bucket_cap_mb`: None
|
| 419 |
+
- `ddp_broadcast_buffers`: False
|
| 420 |
+
- `dataloader_pin_memory`: True
|
| 421 |
+
- `dataloader_persistent_workers`: False
|
| 422 |
+
- `skip_memory_metrics`: True
|
| 423 |
+
- `push_to_hub`: False
|
| 424 |
+
- `resume_from_checkpoint`: None
|
| 425 |
+
- `hub_model_id`: None
|
| 426 |
+
- `hub_strategy`: every_save
|
| 427 |
+
- `hub_private_repo`: None
|
| 428 |
+
- `hub_always_push`: False
|
| 429 |
+
- `hub_revision`: None
|
| 430 |
+
- `gradient_checkpointing`: False
|
| 431 |
+
- `gradient_checkpointing_kwargs`: None
|
| 432 |
+
- `include_for_metrics`: []
|
| 433 |
+
- `eval_do_concat_batches`: True
|
| 434 |
+
- `auto_find_batch_size`: False
|
| 435 |
+
- `full_determinism`: False
|
| 436 |
+
- `ddp_timeout`: 1800
|
| 437 |
+
- `torch_compile`: False
|
| 438 |
+
- `torch_compile_backend`: None
|
| 439 |
+
- `torch_compile_mode`: None
|
| 440 |
+
- `include_num_input_tokens_seen`: no
|
| 441 |
+
- `neftune_noise_alpha`: None
|
| 442 |
+
- `optim_target_modules`: None
|
| 443 |
+
- `batch_eval_metrics`: False
|
| 444 |
+
- `eval_on_start`: False
|
| 445 |
+
- `use_liger_kernel`: False
|
| 446 |
+
- `liger_kernel_config`: None
|
| 447 |
+
- `eval_use_gather_object`: False
|
| 448 |
+
- `average_tokens_across_devices`: True
|
| 449 |
+
- `use_cache`: False
|
| 450 |
+
- `prompts`: None
|
| 451 |
+
- `batch_sampler`: batch_sampler
|
| 452 |
+
- `multi_dataset_batch_sampler`: proportional
|
| 453 |
+
- `router_mapping`: {}
|
| 454 |
+
- `learning_rate_mapping`: {}
|
| 455 |
+
|
| 456 |
+
</details>
|
| 457 |
+
|
| 458 |
+
### Training Logs
|
| 459 |
+
| Epoch | Step | Training Loss | Validation Loss | trivia_qa_eval_cosine_accuracy |
|
| 460 |
+
|:----------:|:--------:|:-------------:|:---------------:|:------------------------------:|
|
| 461 |
+
| 0.0011 | 1 | 0.7126 | - | - |
|
| 462 |
+
| 0.0539 | 50 | 0.7164 | - | - |
|
| 463 |
+
| 0.1079 | 100 | 0.7126 | - | - |
|
| 464 |
+
| 0.1618 | 150 | 0.6888 | - | - |
|
| 465 |
+
| 0.2157 | 200 | 0.6802 | - | - |
|
| 466 |
+
| 0.2697 | 250 | 0.6422 | - | - |
|
| 467 |
+
| 0.3236 | 300 | 0.6562 | - | - |
|
| 468 |
+
| 0.3776 | 350 | 0.6356 | - | - |
|
| 469 |
+
| 0.4315 | 400 | 0.6532 | - | - |
|
| 470 |
+
| 0.4854 | 450 | 0.6106 | - | - |
|
| 471 |
+
| 0.5394 | 500 | 0.6104 | 0.5472 | 0.7970 |
|
| 472 |
+
| 0.5933 | 550 | 0.6301 | - | - |
|
| 473 |
+
| 0.6472 | 600 | 0.6259 | - | - |
|
| 474 |
+
| 0.7012 | 650 | 0.5759 | - | - |
|
| 475 |
+
| 0.7551 | 700 | 0.6089 | - | - |
|
| 476 |
+
| 0.8091 | 750 | 0.5835 | - | - |
|
| 477 |
+
| 0.8630 | 800 | 0.5890 | - | - |
|
| 478 |
+
| 0.9169 | 850 | 0.5577 | - | - |
|
| 479 |
+
| 0.9709 | 900 | 0.5569 | - | - |
|
| 480 |
+
| 1.0248 | 950 | 0.5427 | - | - |
|
| 481 |
+
| 1.0787 | 1000 | 0.4698 | 0.5046 | 0.8190 |
|
| 482 |
+
| 1.1327 | 1050 | 0.4662 | - | - |
|
| 483 |
+
| 1.1866 | 1100 | 0.4634 | - | - |
|
| 484 |
+
| 1.2406 | 1150 | 0.4597 | - | - |
|
| 485 |
+
| 1.2945 | 1200 | 0.4585 | - | - |
|
| 486 |
+
| 1.3484 | 1250 | 0.5140 | - | - |
|
| 487 |
+
| 1.4024 | 1300 | 0.4542 | - | - |
|
| 488 |
+
| 1.4563 | 1350 | 0.4579 | - | - |
|
| 489 |
+
| 1.5102 | 1400 | 0.4910 | - | - |
|
| 490 |
+
| 1.5642 | 1450 | 0.5067 | - | - |
|
| 491 |
+
| 1.6181 | 1500 | 0.4800 | 0.4875 | 0.8300 |
|
| 492 |
+
| 1.6721 | 1550 | 0.4638 | - | - |
|
| 493 |
+
| 1.7260 | 1600 | 0.4760 | - | - |
|
| 494 |
+
| 1.7799 | 1650 | 0.4699 | - | - |
|
| 495 |
+
| 1.8339 | 1700 | 0.4912 | - | - |
|
| 496 |
+
| 1.8878 | 1750 | 0.4726 | - | - |
|
| 497 |
+
| 1.9417 | 1800 | 0.4764 | - | - |
|
| 498 |
+
| 1.9957 | 1850 | 0.4802 | - | - |
|
| 499 |
+
| 2.0496 | 1900 | 0.3941 | - | - |
|
| 500 |
+
| 2.1036 | 1950 | 0.3991 | - | - |
|
| 501 |
+
| **2.1575** | **2000** | **0.4114** | **0.4734** | **0.838** |
|
| 502 |
+
| 2.2114 | 2050 | 0.3981 | - | - |
|
| 503 |
+
| 2.2654 | 2100 | 0.4023 | - | - |
|
| 504 |
+
| 2.3193 | 2150 | 0.3932 | - | - |
|
| 505 |
+
| 2.3732 | 2200 | 0.3887 | - | - |
|
| 506 |
+
| 2.4272 | 2250 | 0.3894 | - | - |
|
| 507 |
+
| 2.4811 | 2300 | 0.3858 | - | - |
|
| 508 |
+
| 2.5351 | 2350 | 0.3907 | - | - |
|
| 509 |
+
| 2.5890 | 2400 | 0.3934 | - | - |
|
| 510 |
+
| 2.6429 | 2450 | 0.3871 | - | - |
|
| 511 |
+
| 2.6969 | 2500 | 0.3763 | 0.4681 | 0.8310 |
|
| 512 |
+
| 2.7508 | 2550 | 0.3997 | - | - |
|
| 513 |
+
| 2.8047 | 2600 | 0.3941 | - | - |
|
| 514 |
+
| 2.8587 | 2650 | 0.3884 | - | - |
|
| 515 |
+
| 2.9126 | 2700 | 0.3771 | - | - |
|
| 516 |
+
| 2.9666 | 2750 | 0.4168 | - | - |
|
| 517 |
+
| 3.0205 | 2800 | 0.3722 | - | - |
|
| 518 |
+
| 3.0744 | 2850 | 0.3565 | - | - |
|
| 519 |
+
| 3.1284 | 2900 | 0.3499 | - | - |
|
| 520 |
+
| 3.1823 | 2950 | 0.3428 | - | - |
|
| 521 |
+
| 3.2362 | 3000 | 0.3583 | 0.4669 | 0.8320 |
|
| 522 |
+
| 3.2902 | 3050 | 0.3444 | - | - |
|
| 523 |
+
| 3.3441 | 3100 | 0.3252 | - | - |
|
| 524 |
+
| 3.3981 | 3150 | 0.3563 | - | - |
|
| 525 |
+
| 3.4520 | 3200 | 0.3465 | - | - |
|
| 526 |
+
| 3.5059 | 3250 | 0.3328 | - | - |
|
| 527 |
+
| 3.5599 | 3300 | 0.3438 | - | - |
|
| 528 |
+
| 3.6138 | 3350 | 0.3330 | - | - |
|
| 529 |
+
| 3.6677 | 3400 | 0.3567 | - | - |
|
| 530 |
+
| 3.7217 | 3450 | 0.3462 | - | - |
|
| 531 |
+
| 3.7756 | 3500 | 0.3435 | 0.4639 | 0.8340 |
|
| 532 |
+
| 3.8296 | 3550 | 0.3532 | - | - |
|
| 533 |
+
| 3.8835 | 3600 | 0.3480 | - | - |
|
| 534 |
+
| 3.9374 | 3650 | 0.3361 | - | - |
|
| 535 |
+
| 3.9914 | 3700 | 0.3628 | - | - |
|
| 536 |
+
|
| 537 |
+
* The bold row denotes the saved checkpoint.
|
| 538 |
+
|
| 539 |
+
### Framework Versions
|
| 540 |
+
- Python: 3.12.12
|
| 541 |
+
- Sentence Transformers: 5.2.2
|
| 542 |
+
- Transformers: 5.0.0
|
| 543 |
+
- PyTorch: 2.9.0+cu128
|
| 544 |
+
- Accelerate: 1.12.0
|
| 545 |
+
- Datasets: 4.0.0
|
| 546 |
+
- Tokenizers: 0.22.2
|
| 547 |
+
|
| 548 |
+
## Citation
|
| 549 |
+
|
| 550 |
+
### BibTeX
|
| 551 |
+
|
| 552 |
+
#### Sentence Transformers
|
| 553 |
+
```bibtex
|
| 554 |
+
@inproceedings{reimers-2019-sentence-bert,
|
| 555 |
+
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
|
| 556 |
+
author = "Reimers, Nils and Gurevych, Iryna",
|
| 557 |
+
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
|
| 558 |
+
month = "11",
|
| 559 |
+
year = "2019",
|
| 560 |
+
publisher = "Association for Computational Linguistics",
|
| 561 |
+
url = "https://arxiv.org/abs/1908.10084",
|
| 562 |
+
}
|
| 563 |
+
```
|
| 564 |
+
|
| 565 |
+
#### MultipleNegativesRankingLoss
|
| 566 |
+
```bibtex
|
| 567 |
+
@misc{henderson2017efficient,
|
| 568 |
+
title={Efficient Natural Language Response Suggestion for Smart Reply},
|
| 569 |
+
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},
|
| 570 |
+
year={2017},
|
| 571 |
+
eprint={1705.00652},
|
| 572 |
+
archivePrefix={arXiv},
|
| 573 |
+
primaryClass={cs.CL}
|
| 574 |
+
}
|
| 575 |
+
```
|
| 576 |
+
|
| 577 |
+
<!--
|
| 578 |
+
## Glossary
|
| 579 |
+
|
| 580 |
+
*Clearly define terms in order to be accessible across audiences.*
|
| 581 |
+
-->
|
| 582 |
+
|
| 583 |
+
<!--
|
| 584 |
+
## Model Card Authors
|
| 585 |
+
|
| 586 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
| 587 |
+
-->
|
| 588 |
+
|
| 589 |
+
<!--
|
| 590 |
+
## Model Card Contact
|
| 591 |
+
|
| 592 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
| 593 |
+
-->
|
config.json
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_cross_attention": false,
|
| 3 |
+
"architectures": [
|
| 4 |
+
"BertModel"
|
| 5 |
+
],
|
| 6 |
+
"attention_probs_dropout_prob": 0.1,
|
| 7 |
+
"bos_token_id": null,
|
| 8 |
+
"classifier_dropout": null,
|
| 9 |
+
"dtype": "float32",
|
| 10 |
+
"eos_token_id": null,
|
| 11 |
+
"gradient_checkpointing": false,
|
| 12 |
+
"hidden_act": "gelu",
|
| 13 |
+
"hidden_dropout_prob": 0.1,
|
| 14 |
+
"hidden_size": 384,
|
| 15 |
+
"initializer_range": 0.02,
|
| 16 |
+
"intermediate_size": 1536,
|
| 17 |
+
"is_decoder": false,
|
| 18 |
+
"layer_norm_eps": 1e-12,
|
| 19 |
+
"max_position_embeddings": 512,
|
| 20 |
+
"model_type": "bert",
|
| 21 |
+
"num_attention_heads": 12,
|
| 22 |
+
"num_hidden_layers": 6,
|
| 23 |
+
"pad_token_id": 0,
|
| 24 |
+
"position_embedding_type": "absolute",
|
| 25 |
+
"tie_word_embeddings": true,
|
| 26 |
+
"transformers_version": "5.0.0",
|
| 27 |
+
"type_vocab_size": 2,
|
| 28 |
+
"use_cache": true,
|
| 29 |
+
"vocab_size": 30522
|
| 30 |
+
}
|
config_sentence_transformers.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"__version__": {
|
| 3 |
+
"sentence_transformers": "5.2.2",
|
| 4 |
+
"transformers": "5.0.0",
|
| 5 |
+
"pytorch": "2.9.0+cu128"
|
| 6 |
+
},
|
| 7 |
+
"model_type": "SentenceTransformer",
|
| 8 |
+
"prompts": {
|
| 9 |
+
"query": "",
|
| 10 |
+
"document": ""
|
| 11 |
+
},
|
| 12 |
+
"default_prompt_name": null,
|
| 13 |
+
"similarity_fn_name": "cosine"
|
| 14 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:856be15cd4795d1d80f180d4697eec76db9e254cc138888618fe678cc0638e4e
|
| 3 |
+
size 90864176
|
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 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"backend": "tokenizers",
|
| 3 |
+
"cls_token": "[CLS]",
|
| 4 |
+
"do_basic_tokenize": true,
|
| 5 |
+
"do_lower_case": true,
|
| 6 |
+
"is_local": false,
|
| 7 |
+
"mask_token": "[MASK]",
|
| 8 |
+
"model_max_length": 256,
|
| 9 |
+
"never_split": null,
|
| 10 |
+
"pad_token": "[PAD]",
|
| 11 |
+
"sep_token": "[SEP]",
|
| 12 |
+
"strip_accents": null,
|
| 13 |
+
"tokenize_chinese_chars": true,
|
| 14 |
+
"tokenizer_class": "BertTokenizer",
|
| 15 |
+
"unk_token": "[UNK]"
|
| 16 |
+
}
|