TomatenMarc commited on
Commit
b72cc54
1 Parent(s): c20ff31

Add new SentenceTransformer model.

Browse files
README.md CHANGED
@@ -52,18 +52,18 @@ Relevant Argument Properties (WRAP) into the embedding space.
52
 
53
  ## Class Semantics
54
 
55
- The model, to some degree, captures the semantics of the critical components of an argument, as defined by the [Cambridge Dictionary](https://dictionary.cambridge.org).
56
  It encodes *inference* as *a guess that one makes or an opinion formed based on available information*, and it also leverages the definition of
57
  *information* as *facts or details about a person, company, product, etc.*.
58
 
59
  Consequently, it has also learned the semantics of:
60
 
61
  * *Statement*, which refers to unique cases where only the *inference* is presented as *something that someone says or writes officially, or an
62
- action done to express an opinion* (see ex. 1).
63
  * *Reason*, which represents a full argument where the *inference* is based on direct *information* mentioned in the tweet, such as a
64
- source-reference or quotation, and thus reveals the author’s motivation *to try to understand and to make judgments based on practical facts* (see ex. 2).
65
- * *Notification*, which refers to a tweet that limits itself to providing *information*, such as media channels promoting their latest articles (see ex. 3).
66
- * *None*, a tweet that provides neither *inference* nor *information* (see ex. 4).
67
 
68
  In its entirety, WRAPresentations encodes the following hierarchy for tweets:
69
 
@@ -199,16 +199,16 @@ WRAPresentation model using the
199
  | Model | Accuracy | Precision | Recall | F1 | Support |
200
  |---------------------|----------|-----------|--------|--------|---------|
201
  | vinai/bertweet-base | 60.62% | 50.08% | 99.89% | 66.71% | 86,142 |
202
- | WRAPresentations | 71.56% | 65.70% | 83.20% | 73.42% | 86,142 |
203
 
204
  An evaluation was conducted on previously unseen data from the holdout topic abortion, resulting in the model achieving a sophisticated macro
205
- F1-score of 71.56%. The recall, which stands at 87.54%, indicates the model's ability to capture subtle tweet patterns and class-specific features for
206
- Reason, Statement, Notification, and None. Despite having a lower precision of 65.70%, the model's primary focus is on prioritizing recall to capture
207
  relevant instances. Fine-tuning precision can be addressed in a subsequent classification phase, when using this model
208
  for `AutoModelForSequenceClassification`. In contrast, the baseline model (*vinai/bertweet-base*) achieved an exceptional recall of 99.89%, but it
209
  comes with a precision trade-off (50.08%), possibly indicating overfitting. However, WRAPresentations demonstrated its ability to effectively
210
  distinguish between tweets of the argument framework, capturing intra-class semantics while discerning inter-class semantics.
211
- This is indicated by its better F1 score of 71.56%, showcasing a superior balance between recall and precision.
212
  As a result, WRAPresentations proves to be more suitable for argument mining on Twitter, as it achieves a more reliable performance in identifying
213
  relevant instances in the data.
214
 
 
52
 
53
  ## Class Semantics
54
 
55
+ WRAPresentations, to some degree, captures the semantics of the critical components of an argument (inference and information), as defined by the[Cambridge Dictionary](https://dictionary.cambridge.org).
56
  It encodes *inference* as *a guess that one makes or an opinion formed based on available information*, and it also leverages the definition of
57
  *information* as *facts or details about a person, company, product, etc.*.
58
 
59
  Consequently, it has also learned the semantics of:
60
 
61
  * *Statement*, which refers to unique cases where only the *inference* is presented as *something that someone says or writes officially, or an
62
+ action done to express an opinion*.
63
  * *Reason*, which represents a full argument where the *inference* is based on direct *information* mentioned in the tweet, such as a
64
+ source-reference or quotation, and thus reveals the author’s motivation *to try to understand and to make judgments based on practical facts*.
65
+ * *Notification*, which refers to a tweet that limits itself to providing *information*, such as media channels promoting their latest articles.
66
+ * *None*, a tweet that provides neither *inference* nor *information*.
67
 
68
  In its entirety, WRAPresentations encodes the following hierarchy for tweets:
69
 
 
199
  | Model | Accuracy | Precision | Recall | F1 | Support |
200
  |---------------------|----------|-----------|--------|--------|---------|
201
  | vinai/bertweet-base | 60.62% | 50.08% | 99.89% | 66.71% | 86,142 |
202
+ | WRAPresentations | 72.25% | 65.45% | 88.21% | 75.14% | 86,142 |
203
 
204
  An evaluation was conducted on previously unseen data from the holdout topic abortion, resulting in the model achieving a sophisticated macro
205
+ F1-score of 75.14%. The recall, which stands at 88.21%, indicates the model's ability to capture subtle tweet patterns and class-specific features for
206
+ Reason, Statement, Notification, and None. Despite having a lower precision of 65.45%, the model's primary focus is on prioritizing recall to capture
207
  relevant instances. Fine-tuning precision can be addressed in a subsequent classification phase, when using this model
208
  for `AutoModelForSequenceClassification`. In contrast, the baseline model (*vinai/bertweet-base*) achieved an exceptional recall of 99.89%, but it
209
  comes with a precision trade-off (50.08%), possibly indicating overfitting. However, WRAPresentations demonstrated its ability to effectively
210
  distinguish between tweets of the argument framework, capturing intra-class semantics while discerning inter-class semantics.
211
+ This is indicated by its better F1 score of 75.14%, showcasing a superior balance between recall and precision.
212
  As a result, WRAPresentations proves to be more suitable for argument mining on Twitter, as it achieves a more reliable performance in identifying
213
  relevant instances in the data.
214
 
binary_classification_evaluation_baseline-heldout_results.csv CHANGED
@@ -1,2 +1,2 @@
1
  epoch,steps,cossim_accuracy,cossim_accuracy_threshold,cossim_f1,cossim_precision,cossim_recall,cossim_f1_threshold,cossim_ap,manhattan_accuracy,manhattan_accuracy_threshold,manhattan_f1,manhattan_precision,manhattan_recall,manhattan_f1_threshold,manhattan_ap,euclidean_accuracy,euclidean_accuracy_threshold,euclidean_f1,euclidean_precision,euclidean_recall,euclidean_f1_threshold,euclidean_ap,dot_accuracy,dot_accuracy_threshold,dot_f1,dot_precision,dot_recall,dot_f1_threshold,dot_ap
2
- -1,-1,0.606235997012696,0.7620757818222046,0.6671155668611901,0.5007862812640408,0.9988797610156833,0.2958066463470459,0.632370163297349,0.6153286034353995,72.18704223632812,0.6850335895837661,0.5444897060442585,0.9233756534727409,87.33160400390625,0.6237689103378412,0.6188946975354742,3.8260855674743652,0.6802268387246109,0.5552144341583223,0.8778939507094847,4.539368152618408,0.6216050521813652,0.5073562359970127,15.957334518432617,0.6667997405060133,0.5006744604316546,0.9979088872292756,9.347124099731445,0.5036378545091198
 
1
  epoch,steps,cossim_accuracy,cossim_accuracy_threshold,cossim_f1,cossim_precision,cossim_recall,cossim_f1_threshold,cossim_ap,manhattan_accuracy,manhattan_accuracy_threshold,manhattan_f1,manhattan_precision,manhattan_recall,manhattan_f1_threshold,manhattan_ap,euclidean_accuracy,euclidean_accuracy_threshold,euclidean_f1,euclidean_precision,euclidean_recall,euclidean_f1_threshold,euclidean_ap,dot_accuracy,dot_accuracy_threshold,dot_f1,dot_precision,dot_recall,dot_f1_threshold,dot_ap
2
+ -1,-1,0.606235997012696,0.7620757818222046,0.6671155668611901,0.5007862812640408,0.9988797610156833,0.2958066463470459,0.6323701642846189,0.6153286034353995,72.18704223632812,0.6850335895837661,0.5444897060442585,0.9233756534727409,87.33160400390625,0.6237689041144366,0.6188946975354742,3.8260855674743652,0.6802268387246109,0.5552144341583223,0.8778939507094847,4.539368152618408,0.6216050521813652,0.5073562359970127,15.957334518432617,0.6667997405060133,0.5006744604316546,0.9979088872292756,9.347124099731445,0.5036378539343036
binary_classification_evaluation_fine-tune-heldout_results.csv CHANGED
@@ -1,2 +1,2 @@
1
  epoch,steps,cossim_accuracy,cossim_accuracy_threshold,cossim_f1,cossim_precision,cossim_recall,cossim_f1_threshold,cossim_ap,manhattan_accuracy,manhattan_accuracy_threshold,manhattan_f1,manhattan_precision,manhattan_recall,manhattan_f1_threshold,manhattan_ap,euclidean_accuracy,euclidean_accuracy_threshold,euclidean_f1,euclidean_precision,euclidean_recall,euclidean_f1_threshold,euclidean_ap,dot_accuracy,dot_accuracy_threshold,dot_f1,dot_precision,dot_recall,dot_f1_threshold,dot_ap
2
- -1,-1,0.7155713218820015,0.5635744333267212,0.7342164228285225,0.6569760584974643,0.8320388349514564,0.4611433148384094,0.7701340554755384,0.7225914861837192,166.821533203125,0.7399493229612176,0.651601716835792,0.8560119492158327,203.3678741455078,0.774587176530623,0.7169716206123973,8.298559188842773,0.7285230528783538,0.6809284207109235,0.7832710978342047,9.337143898010254,0.7708316235108152,0.7108849887976102,57.948116302490234,0.7276492693110648,0.6582263181749509,0.8134428678117999,41.91051483154297,0.7366228479917187
 
1
  epoch,steps,cossim_accuracy,cossim_accuracy_threshold,cossim_f1,cossim_precision,cossim_recall,cossim_f1_threshold,cossim_ap,manhattan_accuracy,manhattan_accuracy_threshold,manhattan_f1,manhattan_precision,manhattan_recall,manhattan_f1_threshold,manhattan_ap,euclidean_accuracy,euclidean_accuracy_threshold,euclidean_f1,euclidean_precision,euclidean_recall,euclidean_f1_threshold,euclidean_ap,dot_accuracy,dot_accuracy_threshold,dot_f1,dot_precision,dot_recall,dot_f1_threshold,dot_ap
2
+ -1,-1,0.7224607916355489,0.7997592687606812,0.7514314798320398,0.6544940707081902,0.8820761762509335,0.468674898147583,0.7656845929850138,0.7232449589245705,122.13343048095703,0.7540637796252636,0.6449113870317309,0.9076923076923077,204.0776824951172,0.7707895435659883,0.7226661687826736,6.091953277587891,0.7526959588167692,0.6506627473394845,0.8926811053024645,9.70598030090332,0.7693129191511483,0.7146191187453323,64.32061767578125,0.7447954113342601,0.6508978189840543,0.8703510082150859,41.77745819091797,0.7333139687176903
eval/binary_classification_evaluation_fine-tune-test_results.csv CHANGED
@@ -1,31 +1,31 @@
1
  epoch,steps,cossim_accuracy,cossim_accuracy_threshold,cossim_f1,cossim_precision,cossim_recall,cossim_f1_threshold,cossim_ap,manhattan_accuracy,manhattan_accuracy_threshold,manhattan_f1,manhattan_precision,manhattan_recall,manhattan_f1_threshold,manhattan_ap,euclidean_accuracy,euclidean_accuracy_threshold,euclidean_f1,euclidean_precision,euclidean_recall,euclidean_f1_threshold,euclidean_ap,dot_accuracy,dot_accuracy_threshold,dot_f1,dot_precision,dot_recall,dot_f1_threshold,dot_ap
2
- 0,1000,0.7678939453016209,0.5616649389266968,0.7749335063250081,0.725400179746897,0.8317272879184537,0.4328032433986664,0.8487791808297954,0.7691472177351975,175.79025268554688,0.7765562172341834,0.7149149524389673,0.849830111959004,196.30166625976562,0.8516339941615735,0.7659165599064224,8.315374374389648,0.7702681429396022,0.7280888844799365,0.8176349356653484,9.056646347045898,0.8479770494283996,0.7686737592602908,41.60350799560547,0.7689403869344098,0.7262912890605655,0.8169108227037264,32.74320602416992,0.817298703547951
3
- 0,2000,0.7677825433075252,0.5190510153770447,0.7758532721071993,0.7490407549517785,0.8046566033532,0.5155777335166931,0.8368960901274767,0.7638973987634379,170.4642333984375,0.770553674954282,0.721071922712819,0.8273269091516738,193.61415100097656,0.8367475169414185,0.7642594552442489,7.980064392089844,0.7707439294186635,0.7390104273154774,0.8053250153177742,8.939167976379395,0.8344096422873637,0.7718347908427561,42.945762634277344,0.7775990048809508,0.7556431480751544,0.8008689355539464,42.44209671020508,0.8302310576373584
4
- 0,3000,0.7739375034813123,0.5127902030944824,0.7809338469536009,0.7488815604468645,0.8158525037598173,0.49300330877304077,0.8449476727696611,0.7766668523366568,200.49417114257812,0.7910268186346171,0.738614674687734,0.8514454408733916,201.90072631835938,0.8522992953865132,0.7769035815741102,9.25938606262207,0.7833858384824435,0.7612665878333991,0.8068289422380661,9.25938606262207,0.8462070606506461,0.7681306745390742,47.408573150634766,0.765128667662561,0.7744493894784891,0.7560296329304295,47.133384704589844,0.8077331359909563
5
- 0,4000,0.7674901130730241,0.6995731592178345,0.767450599287334,0.7175846276866413,0.8247646632874729,0.4543057382106781,0.8436346872666978,0.7675875898178577,171.60763549804688,0.7672268583405124,0.7126239101875075,0.8308917729627361,204.24862670898438,0.8449665781190565,0.7665988971202584,7.342679977416992,0.7594777948805655,0.7777453441531905,0.7420486826714198,8.261296272277832,0.8416007984609263,0.7677268423104774,52.90253829956055,0.7744300725479566,0.7217516843118383,0.8354035537236116,37.78709030151367,0.794628873897718
6
- 0,5000,0.7745919901966245,0.6659778356552124,0.7749942891951571,0.7118810034505269,0.8503871219294825,0.45358020067214966,0.8488832109960202,0.7747312426892441,170.88339233398438,0.7777511898114914,0.7440227897039373,0.8146827828218125,199.32559204101562,0.8511201654908707,0.7748983456803876,7.782614707946777,0.7700333358780569,0.7088643178410795,0.8427560853339274,9.623931884765625,0.8480727902398433,0.7755528323956998,58.938602447509766,0.7752517606792602,0.7350657779774832,0.8200857795354537,40.16120529174805,0.7930881406399948
7
- 0,-1,0.7797304071742884,0.67124342918396,0.7751530612244898,0.7150656563279522,0.8462652481479418,0.45725303888320923,0.8486250473756219,0.776569375591823,163.75259399414062,0.7768396302036017,0.7465208237046167,0.8097253940845541,198.85543823242188,0.8510887216883583,0.7787138639781652,7.571657657623291,0.7697543500511771,0.7119325980972215,0.8377986965966691,9.60276985168457,0.8479597311663336,0.7776694702835181,55.622920989990234,0.7724684702882175,0.7064773120886734,0.8520581518409179,38.716949462890625,0.7918274296961403
8
- 1,1000,0.7749261961789116,0.566508412361145,0.7798376850675448,0.7295042204327157,0.8376315936055255,0.4021725654602051,0.8511605513699232,0.7762630201080599,171.59909057617188,0.7799883653286795,0.7278342058915777,0.8401938394697265,213.2512969970703,0.8509098800953989,0.7749401214281736,8.299052238464355,0.7766009754023202,0.7426610751048418,0.813791566869047,9.897568702697754,0.849563100342725,0.7748565699326018,53.404930114746094,0.7776896913280311,0.7189125295508274,0.846933660112516,33.58881378173828,0.8062415345332914
9
- 1,2000,0.7784075084944021,0.6520156264305115,0.7775647551929763,0.7401887979861548,0.8189160585974489,0.4154204726219177,0.8512251193283239,0.7775441430401604,162.30169677734375,0.7795258085187075,0.7347945070387811,0.8300562580070183,212.55264282226562,0.8508408487911324,0.7779619005180193,7.799623012542725,0.7731482731617746,0.7503144654088051,0.7974154737369799,9.854764938354492,0.8498399451414931,0.7787556397259511,54.70463562011719,0.7773618032067335,0.7245163153335259,0.8385228095582911,34.559288024902344,0.8053294570747453
10
- 1,3000,0.7781568540076867,0.6612838506698608,0.775138972607327,0.7467031407262122,0.8058263242912048,0.4267839193344116,0.8503063748678845,0.7782682560017824,162.04498291015625,0.7756105992315158,0.7420561223191798,0.812343340945803,211.3235321044922,0.8499273517817276,0.7778922742717095,7.748963832855225,0.7710258717046972,0.7550726185390859,0.7876677992536066,9.824663162231445,0.8491794467884947,0.778449284242188,56.12619400024414,0.7765954623855644,0.7415872694100952,0.8150726898011474,36.294673919677734,0.8037081919007742
11
- 1,4000,0.7214393137637164,0.5044573545455933,0.6962221636964856,0.7334155363748459,0.6626190608811898,0.45556509494781494,0.7796048578772816,0.7197265081044951,180.50286865234375,0.6977414421832725,0.7390816771540714,0.6607809279786108,199.2108917236328,0.7812933138616089,0.7160502422993371,8.91546630859375,0.6965976246485742,0.718189884649512,0.6762658051579123,9.800008773803711,0.7813535500643046,0.721216509775525,42.53407669067383,0.6979591222872268,0.7586725519045283,0.6462429677491227,41.97949981689453,0.7227403664766267
12
- 1,5000,0.7312287639948755,0.6148228049278259,0.7150222843659879,0.7494504152418173,0.6836183367682281,0.45740455389022827,0.7955877922714433,0.7304768005347295,198.79611206054688,0.7168404934562157,0.7446578631452581,0.6910265693755918,201.54525756835938,0.7955900503582694,0.73104773575447,8.091403007507324,0.7107716434815439,0.7425364077669903,0.6816131008745057,9.69920539855957,0.7944885709600145,0.7335542806216231,44.558616638183594,0.713370354549338,0.7298100454492483,0.6976549880242856,39.405242919921875,0.726060786237057
13
- 1,-1,0.7382888653706902,0.522275984287262,0.7203785844620153,0.7707202488651874,0.6762101041608645,0.48286569118499756,0.790454845458989,0.7378989583913552,190.44403076171875,0.7190101103351747,0.7725338367516719,0.6724224363616109,195.2378692626953,0.7945392109799041,0.7367431627026124,9.284839630126953,0.7173958488875616,0.7733067216070049,0.6690246755416922,9.304485321044922,0.7916372746891804,0.7394028853116471,44.70606994628906,0.7188546877581985,0.7803398454062164,0.6663510276833955,43.997772216796875,0.7269551575769607
14
- 2,1000,0.7374254999164485,0.49638476967811584,0.7150868144060933,0.7745225412498375,0.6641229878014816,0.4762795865535736,0.7895789121471601,0.739082604578622,193.4174346923828,0.7187663965487088,0.753928942701645,0.6867375926029076,201.81869506835938,0.7920154306606293,0.7373001726730909,9.189388275146484,0.7162266969235631,0.7394424673784105,0.6944243301955105,9.865802764892578,0.7926948853939454,0.7390965298278839,45.12528991699219,0.7149767272063521,0.7877061268266524,0.6545424163092519,45.09357452392578,0.7248229185549959
15
- 2,2000,0.7397510165431961,0.49059003591537476,0.7174882469426934,0.7846067378583,0.6609480309697544,0.49059003591537476,0.7893193607026283,0.7402662507658887,187.76220703125,0.7204597431536973,0.7606017458057327,0.6843424497298501,201.83572387695312,0.7917386019095046,0.7384141926140478,9.029869079589844,0.7164570995127929,0.7522053418279834,0.6839525427505152,9.867085456848145,0.793468666491425,0.7383027906199521,44.93938446044922,0.7148524373384624,0.7762548136544613,0.6624519578900462,42.35888671875,0.7258325223417688
16
- 2,3000,0.7339163371024341,0.5524381399154663,0.7096153846153846,0.7704051673517323,0.6577173731409792,0.4623206853866577,0.7955165384790114,0.7348911045507714,195.80374145507812,0.7160265716422507,0.7561557283117044,0.6799420709630702,200.68942260742188,0.7945675954196826,0.7338467108561243,8.778763771057129,0.7130119479284314,0.7643535334225451,0.6681334595889267,9.822650909423828,0.7944429400411687,0.7333732523812176,44.763912200927734,0.7049359695489359,0.7890978092116612,0.6369966022391801,44.763912200927734,0.7288963131428454
17
- 2,4000,0.7330390463989306,0.5843002796173096,0.7106344203277415,0.7663756162000194,0.6624519578900462,0.45505326986312866,0.7942846131264139,0.7349468055478193,196.8582763671875,0.71410343906709,0.7640970281940563,0.6702500974767448,200.70700073242188,0.793536220750753,0.7333175513841698,8.655280113220215,0.7148661532819949,0.7551520034708233,0.6786609480309698,9.962120056152344,0.7943300826887122,0.7332897008856458,52.494140625,0.7017251489059654,0.7867820069204152,0.6332646354369743,44.95760726928711,0.7309583504671321
18
- 2,5000,0.7313262407397092,0.5530322790145874,0.7062788438183144,0.7762797837549222,0.6478582966635102,0.4647907316684723,0.7913764795949331,0.7316186709742104,195.1563720703125,0.7083424599523956,0.74455871066769,0.6754859911992425,201.52902221679688,0.7906407744746664,0.7307413802707069,9.934350967407227,0.7103064894913642,0.7616345786051256,0.66545981173063,9.974112510681152,0.7927391316736574,0.7327326909151674,52.273109436035156,0.702691850318914,0.7886982145952505,0.6335988414192614,45.618186950683594,0.7295537869972409
19
- 2,-1,0.7312426892441375,0.5550357103347778,0.7061465398212633,0.7756408119729342,0.6480811006517017,0.46514391899108887,0.7913475580426201,0.7318693254609258,195.03797912597656,0.7077027565739934,0.7531031015303397,0.6674650476243524,200.6476593017578,0.7906456213707942,0.7309920347574221,9.92496109008789,0.7107233776480921,0.763473532704302,0.6647913997660558,9.963253021240234,0.7925669002708331,0.7326909151673815,52.270423889160156,0.7026601411125348,0.7883596050580287,0.6337659444104049,45.76286315917969,0.7294773211363561
20
- 3,1000,0.7606528156854008,0.5222647190093994,0.7531634240709377,0.7448054246888701,0.7617111346293098,0.45683982968330383,0.8206757503089424,0.7610148721662118,183.16404724121094,0.7484159945569745,0.7620587131599458,0.7352531610315824,199.61854553222656,0.8093893305455506,0.7592742160084666,9.634052276611328,0.7524050488157011,0.7728994743179348,0.7329694201526207,9.713886260986328,0.8228600831850887,0.7607502924302345,47.358665466308594,0.7518852833638027,0.7716798592788039,0.7330808221467164,44.067352294921875,0.7666417639308817
21
- 3,2000,0.733582131120147,0.48442283272743225,0.7355212850581201,0.6818917118660196,0.7983066896897455,0.44973790645599365,0.8050704151130821,0.7298083885701554,201.81997680664062,0.7293821485251464,0.7168289141074597,0.7423828886537069,207.67794799804688,0.7973270462595166,0.7338467108561243,9.741870880126953,0.7349640213947642,0.6812183465297097,0.7979167827104106,10.044559478759766,0.8076746774003403,0.7311591377485657,46.107749938964844,0.735662325546212,0.672056757192546,0.8125661449339944,40.949432373046875,0.7612766515277805
22
- 3,3000,0.7564891661560742,0.48917582631111145,0.7576617553680882,0.708479751835785,0.8141814738483819,0.433296799659729,0.8155651684987558,0.7578538405837465,193.57858276367188,0.7464613425188873,0.7770278413884537,0.7182086559349412,205.32235717773438,0.8003775028938475,0.7567119701442656,9.687872886657715,0.7545677701850991,0.7205960617349654,0.791901075029243,10.162216186523438,0.8194534741932576,0.7561549601737871,43.90623474121094,0.7536313854538912,0.699818563789152,0.8164095137302958,39.590248107910156,0.7722236131029006
23
- 3,4000,0.7385255946081435,0.49491021037101746,0.7461200275438319,0.7112345367331482,0.7846042444159751,0.42813336849212646,0.81639852613666,0.7406283072466997,205.55908203125,0.7280786166273001,0.7640146878824969,0.695371247145324,206.00106811523438,0.8012519667482616,0.7399459700328636,9.668192863464355,0.7449076209667629,0.6602255121363402,0.8545089957110232,10.443655014038086,0.8169782426067075,0.7380382108839748,43.70752716064453,0.7412867436077281,0.683714037027453,0.8094468890993148,38.88163375854492,0.7719869405881279
24
- 3,5000,0.7368963404444939,0.4928392469882965,0.7346872680655826,0.7126364564517396,0.7581462708182476,0.45811888575553894,0.8043990044525555,0.7428424218793517,191.797607421875,0.7264938326316567,0.7735380162949448,0.6848437587032807,193.6221466064453,0.7963762082998681,0.7367292374533504,9.465219497680664,0.738096524887856,0.7167900068230725,0.7607085166824487,9.99488639831543,0.8096146238710167,0.73686848994597,47.051029205322266,0.7248700356433412,0.733456878118318,0.716481925026458,43.94330978393555,0.7609198118503129
25
- 3,-1,0.7370355929371136,0.49309322237968445,0.7345518311189047,0.7122341677783881,0.7583133738093912,0.4582207500934601,0.8044314208625994,0.7428424218793517,191.74285888671875,0.7265364036773183,0.7740614764424288,0.6845095527209937,193.55892944335938,0.796406941812556,0.7366178354592547,9.466680526733398,0.7386227100292954,0.7167448319228653,0.7618782376204534,9.996031761169434,0.8097048652084639,0.736910265693756,48.42571258544922,0.7253521126760564,0.7337436598848807,0.7171503369910321,43.963096618652344,0.761129156358665
26
- 4,1000,0.7396396145491004,0.4853218197822571,0.7335013431554459,0.7294007490636704,0.7376483039046399,0.4670202136039734,0.8048413129762076,0.7425917673926363,192.4800567626953,0.7270748459400233,0.7703431296163555,0.688408622514343,194.97476196289062,0.7964549767767972,0.7383445663677379,9.929280281066895,0.7396969404497635,0.7324759018049752,0.747061772405726,9.958412170410156,0.8107329426732655,0.737634378655378,54.005027770996094,0.7272314910626958,0.740509800525389,0.7144209881356877,44.62757110595703,0.7618423960334315
27
- 4,2000,0.7411296162201303,0.4963013529777527,0.7339127573821732,0.7226781857451404,0.7455021444883864,0.46599581837654114,0.8046479840463622,0.7431070016153289,191.53518676757812,0.7273531493276716,0.7727927815026004,0.686960396591099,194.13693237304688,0.7963775697420389,0.7391800813234557,9.577247619628906,0.7395142578634486,0.7292518480409412,0.7500696262463098,9.973331451416016,0.8109155085919635,0.7393750348131232,54.30863571166992,0.7277907058740569,0.737221063458956,0.7185985629142762,44.826446533203125,0.7611387981164628
28
- 4,3000,0.7421183089177297,0.5031415820121765,0.732177798748248,0.7296766520067491,0.734696151061104,0.47203877568244934,0.8043563389232249,0.7445830780370969,192.44961547851562,0.7284195563978798,0.7776302478502782,0.6850665626914721,192.44961547851562,0.7962850912010971,0.7404751295048181,9.603111267089844,0.740485187010569,0.7282029188432333,0.7531888820809892,9.98196792602539,0.8114153743268732,0.7393750348131232,55.02947235107422,0.7268186075554168,0.7409722222222223,0.713195566200635,45.50734329223633,0.7597365088009872
29
- 4,4000,0.7426613936389461,0.5002992153167725,0.7332384699087636,0.7345706618962433,0.7319111012087116,0.47579818964004517,0.8036745730162046,0.7440121428173564,192.37493896484375,0.7282555718020294,0.7752586396654193,0.6866261906088119,193.13973999023438,0.7962240589563083,0.7407118587422715,9.621297836303711,0.7425506067348249,0.7281488421898006,0.7575335598507213,9.983221054077148,0.8113257016732922,0.7400991477747452,52.59965515136719,0.7278388859255841,0.7544305310660172,0.7030579847379268,46.75685119628906,0.7601106825613864
30
- 4,5000,0.7427449451345179,0.5027592182159424,0.7334051545537417,0.7274769837790442,0.739430735810171,0.47258734703063965,0.8035390497206156,0.7440399933158803,192.66281127929688,0.7285880911251864,0.7753024351924588,0.6871832005792904,192.68093872070312,0.7964492468636172,0.7402662507658887,9.634147644042969,0.7427504124118938,0.7275058088294207,0.7586475797916783,9.984941482543945,0.8113731356681799,0.7408511112348911,53.53934097290039,0.728091314448982,0.7554876770580661,0.702612376761544,47.05971145629883,0.7598655772877093
31
- 4,-1,0.7427449451345179,0.5027490258216858,0.7334116833736606,0.7282462450918477,0.7386509218515012,0.472969651222229,0.8035469606946056,0.7440539185651424,192.65733337402344,0.7286203638081739,0.7753755263654076,0.6871832005792904,192.67977905273438,0.7964520586007977,0.7402384002673648,9.634243965148926,0.7427612825793428,0.7277829747427502,0.758369074806439,9.984170913696289,0.8113810178372788,0.7408511112348911,53.53996658325195,0.7281280212406748,0.7555023207066927,0.7026680777585919,47.05809020996094,0.7598733766921927
 
1
  epoch,steps,cossim_accuracy,cossim_accuracy_threshold,cossim_f1,cossim_precision,cossim_recall,cossim_f1_threshold,cossim_ap,manhattan_accuracy,manhattan_accuracy_threshold,manhattan_f1,manhattan_precision,manhattan_recall,manhattan_f1_threshold,manhattan_ap,euclidean_accuracy,euclidean_accuracy_threshold,euclidean_f1,euclidean_precision,euclidean_recall,euclidean_f1_threshold,euclidean_ap,dot_accuracy,dot_accuracy_threshold,dot_f1,dot_precision,dot_recall,dot_f1_threshold,dot_ap
2
+ 0,1000,0.7664317941291149,0.486582487821579,0.7803905319314238,0.7292483422874014,0.8392469225199131,0.4408458471298218,0.8359886478046229,0.764565810728012,189.147216796875,0.7799080846656742,0.7286581678318295,0.838912716537626,193.72543334960938,0.8393917121665209,0.7655127276778254,8.783966064453125,0.7800096990734833,0.7198718552718364,0.8511112348911045,9.466306686401367,0.8391931466456664,0.7670166545981173,36.696746826171875,0.7787188919580557,0.7347150515056446,0.828329527098535,34.698734283447266,0.8128973550679109
3
+ 0,2000,0.7393750348131232,0.5573605298995972,0.735912426918771,0.7304705720949376,0.7414359717038935,0.4102650582790375,0.8076656543778584,0.7434551328468779,199.37057495117188,0.7488441027070165,0.6868043776285522,0.8232050353701331,220.32708740234375,0.8211288020557157,0.7424107391522308,8.511497497558594,0.7400637331344498,0.7211607072068501,0.7599844037208267,9.90682601928711,0.8160996854852789,0.7356848437587032,39.38921356201172,0.7262602993343414,0.7372528765315199,0.7155907090736924,35.134132385253906,0.7802658061437526
4
+ 0,3000,0.786456302567816,0.7374271154403687,0.7835469018694181,0.7803436274238992,0.7867765833008411,0.4987448751926422,0.8528158825569336,0.7859132178465994,146.7528076171875,0.7966924691850955,0.7410566670660117,0.8613602183479084,207.1959228515625,0.8659377973406632,0.7862891995766724,7.01038122177124,0.7862018654577667,0.7385579323510696,0.8404166434579179,9.810394287109375,0.859054868144066,0.7848270484041664,58.307464599609375,0.7802126519446089,0.7902193784277879,0.7704561911658219,44.82365036010742,0.8401202917713289
5
+ 0,4000,0.7870829387846042,0.5717326402664185,0.7865554398821032,0.7643271934203957,0.8101153010638891,0.47935885190963745,0.8528562703075022,0.7898540633877347,189.68896484375,0.7965073721635471,0.7504803192275482,0.8485489890269036,204.52267456054688,0.8630952748863835,0.7867765833008411,8.835192680358887,0.7864319242511868,0.7604629990896086,0.8142371748454297,9.588624954223633,0.8571658565213038,0.7853979836239069,49.12046813964844,0.7791595317631865,0.7853989813242784,0.7730184370300228,45.11640930175781,0.8364191012827793
6
+ 0,5000,0.7808026513674595,0.5821325778961182,0.7869083690314492,0.7408882986572229,0.8390241185317218,0.5424697399139404,0.8576738380657107,0.7815685400768674,170.3080291748047,0.7900655545488462,0.7573698462167271,0.8257115802372863,184.01321411132812,0.8605206322801829,0.7773770400490169,8.039061546325684,0.7895610789325732,0.7441161248983414,0.8409179524313485,8.928276062011719,0.8591663824472929,0.7778783490224475,53.91118621826172,0.7809087376559339,0.7524854744996772,0.8115635269871331,49.15049362182617,0.817047450871158
7
+ 0,-1,0.7717373140979223,0.5294360518455505,0.7723978113936273,0.7181881299528359,0.8354592547206595,0.4505099058151245,0.8461767046437717,0.7718347908427561,191.7567901611328,0.7739183267189782,0.7221538684229573,0.8336768228151283,206.53079223632812,0.8519611394734723,0.7679914220464547,8.787954330444336,0.7706761375058635,0.7427729985274742,0.8007575335598507,9.45853042602539,0.8473283809397187,0.7697320782041999,47.89080810546875,0.7718713009792317,0.7464488266819294,0.7990865036484153,41.52181625366211,0.8107323333475928
8
+ 1,1000,0.7610148721662118,0.47341352701187134,0.7708736052045699,0.7393427950389976,0.8052136133236785,0.4720102548599243,0.8393056786258988,0.7629783323121484,207.37928771972656,0.778014996875651,0.7304326570520655,0.8322285968918843,207.80984497070312,0.8458845400300564,0.7596919734863254,9.571844100952148,0.7671932773109243,0.7416361225922171,0.7945747228875397,9.65745735168457,0.8408814907414719,0.7610009469169499,48.85084915161133,0.7655571635311142,0.7377201590826921,0.7955773408344009,42.320281982421875,0.7889608097627274
9
+ 1,2000,0.7617668356263577,0.4831611216068268,0.768227188179508,0.7251953704619646,0.816688018715535,0.447949081659317,0.841306812489196,0.7660975881468278,206.78155517578125,0.7777397710827523,0.7370337123478955,0.8232050353701331,209.63221740722656,0.8465489166235076,0.7623099203475743,9.669540405273438,0.7684107879937484,0.7383391092285971,0.80103603854509,9.81333065032959,0.8418975448599488,0.7622542193505264,44.03800964355469,0.7635499079882333,0.7374484315404375,0.791566869046956,40.96443176269531,0.7916606936932462
10
+ 1,3000,0.7644126329861305,0.32492801547050476,0.764449053102658,0.7493179264965495,0.7802038656491951,0.29830923676490784,0.8328818262411662,0.7600818804656604,227.51390075683594,0.7586775704200741,0.7616267647141374,0.7557511279451902,228.03140258789062,0.8336382918546894,0.7614047791455467,10.477073669433594,0.7653250104711335,0.7432230298895216,0.7887818191945636,10.87778377532959,0.8332309655383566,0.7651924469448003,27.151779174804688,0.7641630781276216,0.7672525127744398,0.7610984236617836,27.068164825439453,0.804377275736198
11
+ 1,4000,0.7659861861527322,0.354583740234375,0.763364736049947,0.7639609483960949,0.762769453573219,0.3361058831214905,0.8362531142900129,0.7672394585863087,227.73187255859375,0.7662296130476496,0.7650701096765237,0.7673926363281903,228.43576049804688,0.8383852543413032,0.7676154403163816,10.506753921508789,0.7660727592981506,0.7459299717670651,0.7873335932713196,10.817108154296875,0.8364420469430429,0.7644822592324403,32.58717346191406,0.7630628494931043,0.7641390867197319,0.7619896396145491,28.88344383239746,0.791439876728664
12
+ 1,5000,0.7680053472957166,0.41714152693748474,0.7577735459580468,0.7925802037403071,0.7258953935275442,0.41644296050071716,0.8336291357428097,0.7651228206984905,221.26913452148438,0.7606010761381694,0.7613864701942398,0.759817300729683,225.83990478515625,0.8357898417051105,0.767141981841475,10.097867965698242,0.7572964892675211,0.790677940418826,0.7266195064891662,10.11166763305664,0.8331590369308912,0.7684370300228374,36.36254119873047,0.758169554599372,0.7929818159703217,0.7262853005068791,36.30207443237305,0.7807937708671169
13
+ 1,-1,0.7680331977942405,0.42106810212135315,0.7577179550506299,0.7919944118808273,0.7262853005068791,0.4176786541938782,0.833636826630449,0.7651088954492286,218.78744506835938,0.7607916428402071,0.7619925683792921,0.7595944967414917,225.50914001464844,0.835775475062599,0.7670166545981173,10.075724601745605,0.7570001161845009,0.7909559939301972,0.7258396925304963,10.078554153442383,0.8331679444742818,0.7684231047735754,36.70421600341797,0.7581744958636831,0.7931254752851711,0.7261738985127834,36.548641204833984,0.7795522122685896
14
+ 2,1000,0.7698852559460815,0.45917731523513794,0.7593709454353622,0.777949811574304,0.7416587756920849,0.42607033252716064,0.8260349945603962,0.770734696151061,213.33164978027344,0.7615102756093676,0.7932772094988081,0.7321896061939509,213.5850830078125,0.8277628265717094,0.7703587144209881,9.798596382141113,0.7612718418791178,0.7836291796898036,0.740154848771793,10.001861572265625,0.8264172100812509,0.7694953489667465,40.91225051879883,0.7576349794358721,0.7952847519902021,0.723388848660391,39.9620246887207,0.7706383596974287
15
+ 2,2000,0.7663203921350192,0.4643022418022156,0.7558404805221057,0.7847805229071719,0.7289589483651757,0.4442211389541626,0.8257776933109247,0.7687573107558625,211.79931640625,0.7573733189416261,0.7951120911429621,0.7230546426781039,212.55064392089844,0.8270907892385867,0.769314320726341,9.791139602661133,0.7578818487909399,0.7950215589737317,0.7240572606249652,9.844770431518555,0.8247968913117689,0.764927867208823,41.210693359375,0.7543912476161798,0.7774789419240431,0.7326352141703336,38.983665466308594,0.7525271090538921
16
+ 2,3000,0.7628947808165766,0.466765820980072,0.7455698841468862,0.8011392729134664,0.6972093800479029,0.4552168548107147,0.8245430158227223,0.7608199186765443,209.82913208007812,0.7435288565065412,0.7902806082458066,0.7019996657940177,212.08395385742188,0.8250487096677921,0.7629922575614103,9.727481842041016,0.744505289544756,0.8018513161700896,0.6948142371748455,9.79130744934082,0.8236187301873519,0.7633821645407453,40.188594818115234,0.7475506445672191,0.7935812818668084,0.7065671475519412,39.79478454589844,0.7663008201127002
17
+ 2,4000,0.7655127276778254,0.4727761149406433,0.751514796709891,0.7919185687847008,0.7150336991032139,0.46665799617767334,0.8256477131993547,0.7635353422826269,210.45877075195312,0.7487110382001406,0.7896558116542051,0.7118030412744388,211.28128051757812,0.827269046116811,0.7630479585584582,9.724624633789062,0.7477670073505726,0.788378411757441,0.7111346293098646,9.776246070861816,0.8241057263805148,0.7646772127221078,42.401153564453125,0.7535693557748113,0.78657539225783,0.7232217456692475,41.692344665527344,0.7660479312291009
18
+ 2,5000,0.7624073970924079,0.47480508685112,0.7462254395036195,0.7946136420840675,0.7033921907202139,0.4719405770301819,0.8260047209317134,0.761293377151451,209.33734130859375,0.7438429506246282,0.7981106784962022,0.6964852670862809,211.26072692871094,0.8280162151640978,0.7622681445997883,9.641020774841309,0.7443402425183681,0.8024341554510915,0.6940901242132235,9.752893447875977,0.8247934752022039,0.7636049685289367,43.243228912353516,0.7480436329143941,0.7994677817905341,0.7028351807497354,42.909053802490234,0.769666046201166
19
+ 2,-1,0.7624491728401939,0.4863702952861786,0.7464539007092199,0.7949896141499339,0.7035035927143096,0.47535228729248047,0.8259985722960899,0.7613769286470228,208.98150634765625,0.744395808462661,0.7905462513695414,0.7033364897231661,211.2559814453125,0.8280516075885992,0.7623795465938841,9.617895126342773,0.7443840151336013,0.7929970401158764,0.7013869548264914,9.760398864746094,0.8247848981030891,0.7635771180304127,43.704925537109375,0.748010767000917,0.7975274378705689,0.7042834066729794,43.20008850097656,0.7703023924808915
20
+ 3,1000,0.7468807441653206,0.5050731897354126,0.7435443173887367,0.7175794000310864,0.7714588091126832,0.37026703357696533,0.7977073238610843,0.7473263521417033,214.29537963867188,0.74179991799918,0.686915666081344,0.8062162312705398,228.963134765625,0.8000499172101474,0.7465326129337715,9.728931427001953,0.7454636754746685,0.7270032030071207,0.7648860914610371,10.620023727416992,0.7970713194440227,0.7473820531387512,48.07101821899414,0.7426577414493069,0.7245846912010174,0.7616554336322621,34.42692565917969,0.7280201261344894
21
+ 3,2000,0.7458781262184593,0.720092236995697,0.7463349836976211,0.6588350673716528,0.8606361053862864,0.33686375617980957,0.7983174089777652,0.7465883139308194,151.50115966796875,0.7379131043447827,0.6693727606694181,0.8220910154291762,231.5017547607422,0.7976659280207756,0.746351584693366,7.285037040710449,0.7399749769358119,0.6773559149487518,0.8153511947863866,10.80156135559082,0.7973673654366333,0.7448476577730742,67.33314514160156,0.7461720403336238,0.6746353322528363,0.8346794407619896,31.5607967376709,0.7233088294933525
22
+ 3,3000,0.7482314933437308,0.6788135170936584,0.7427231408954645,0.6561445474349664,0.8556230156519802,0.34727874398231506,0.8052624958972235,0.74779981061661,165.0928955078125,0.7345990948793393,0.697956625297731,0.7753021779089846,221.5665283203125,0.8039966803444085,0.748175792346683,8.29865837097168,0.7363489223954341,0.7510426320667285,0.7222191277223863,10.29574966430664,0.8046418523555745,0.7481061661003732,64.63654327392578,0.7439664980271599,0.657890230327939,0.8559572216342672,31.514062881469727,0.7251012559777354
23
+ 3,4000,0.7483568205870885,0.6468403339385986,0.7433181305477907,0.6639362214726883,0.8442600122542193,0.36670345067977905,0.8100511634526301,0.7484821478304462,173.53329467773438,0.7363483403864783,0.6788887321957411,0.8044337993650086,228.00086975097656,0.8086621316724223,0.7483985963348744,8.187524795532227,0.736533729151205,0.6915910702496516,0.7877235002506545,10.532373428344727,0.8084176294386471,0.7478555116136579,55.885475158691406,0.7435897435897436,0.6776011362074494,0.8238177463376595,34.712608337402344,0.728199161046412
24
+ 3,5000,0.7487745780649474,0.6286755204200745,0.7385057058781311,0.7619893954204805,0.7164262240294101,0.44652101397514343,0.8121020662991324,0.7481061661003732,179.2566680908203,0.735515468543662,0.6814758499441685,0.798863699660224,225.1893310546875,0.8101274009317954,0.7488024285634712,8.48324203491211,0.7376079452863357,0.6838936873914389,0.8004790285746115,10.483926773071289,0.8106823978460422,0.747813735865872,59.67631149291992,0.7426636290868789,0.6656659087397073,0.8398039324903915,35.37864303588867,0.728731711855164
25
+ 3,-1,0.7486910265693756,0.6259658336639404,0.7381954585803052,0.7595757218621096,0.7179858519467498,0.4472625255584717,0.8120948663631773,0.748189717595945,170.84100341796875,0.7354717900704124,0.6816442003661175,0.7985294936779368,224.99334716796875,0.8101333321025699,0.7488302790619952,8.504928588867188,0.7363695660514905,0.6808409999290495,0.801760151506712,10.484764099121094,0.8106463067737346,0.7479947641062775,59.872596740722656,0.7422178509958199,0.6643923053220055,0.8406951484431572,35.47296905517578,0.7286414023745557
26
+ 4,1000,0.7501392524926196,0.4743134081363678,0.7407087543541468,0.7665038131553861,0.7165933270205537,0.46892672777175903,0.8117229912824062,0.7502367292374533,206.0076141357422,0.7381509867512106,0.772424288540557,0.7067899515401326,206.4764404296875,0.8105447519629301,0.7483568205870885,9.838419914245605,0.7386380232882351,0.705792798964756,0.7746894669414582,10.05587387084961,0.8099282781580126,0.7487328023171614,43.34110641479492,0.741543366711503,0.670706979678277,0.8291093410572049,38.03354263305664,0.7267622020485439
27
+ 4,2000,0.7494012142817357,0.4709709882736206,0.7406259861720744,0.7636062470421202,0.7189884698936111,0.4665965735912323,0.811504972240535,0.749637943519189,206.9192352294922,0.7386116191969341,0.771376591873863,0.7085166824486159,207.0909881591797,0.8110615845624107,0.7487049518186375,8.368803977966309,0.7382044173488024,0.698753824341467,0.7823762045340612,10.136795043945312,0.8099362130031903,0.7473681278894893,43.181793212890625,0.7401300775901415,0.7585218967432614,0.7226090347017211,42.75250244140625,0.7536420262577872
28
+ 4,3000,0.7456970979780538,0.6712411642074585,0.7533495660860334,0.6828131012471989,0.8401381384726787,0.43679845333099365,0.8202066972566091,0.7461009302066507,154.74600219726562,0.740471767942684,0.7426394374877441,0.7383167158692141,211.21031188964844,0.8179704159239546,0.7456553222302679,7.134251594543457,0.7422293019646286,0.7260474655728099,0.7591488887651089,10.091227531433105,0.8141645843929346,0.7449312092686459,42.60676574707031,0.7561614824619458,0.7205125618000202,0.7955216398373531,41.880393981933594,0.761104694810306
29
+ 4,4000,0.747423828886537,0.7292289733886719,0.7538784906415775,0.6843850801871791,0.8390798195287695,0.44124242663383484,0.8217723055013232,0.7481340165988971,149.73931884765625,0.742925222510337,0.7478201980868535,0.7380939118810227,210.6751708984375,0.8195341124481053,0.7479669136077536,7.131768226623535,0.7438380760650743,0.7269875033235842,0.7614883306411184,10.072151184082031,0.8155202589056434,0.746351584693366,42.63652038574219,0.7589479112944816,0.7065828011715561,0.8196958725561188,41.66033172607422,0.7617849838312489
30
+ 4,5000,0.7481200913496352,0.7397788763046265,0.7532934131736527,0.6822434130248113,0.8408622514343007,0.4417516589164734,0.8223604832904836,0.749637943519189,209.40219116210938,0.7444073172113043,0.7570992454351708,0.732133905196903,210.01272583007812,0.8202572748893633,0.7480922408511113,7.069145202636719,0.7441329266424322,0.7442676845067283,0.7439982175680945,10.023733139038086,0.8162677044161655,0.747785885367348,42.620033264160156,0.7599622089637231,0.7098474408258988,0.8176906366623963,41.85185241699219,0.7626665612311252
31
+ 4,-1,0.7481340165988971,0.739820659160614,0.7532840159179651,0.6822279968365157,0.8408622514343007,0.44175392389297485,0.8223617033677332,0.749637943519189,209.40499877929688,0.7444033813346926,0.7571506754615894,0.7320782041998551,210.01052856445312,0.8202571976199776,0.7480922408511113,7.069060325622559,0.7441471804799242,0.7442404657770845,0.7440539185651424,10.023887634277344,0.8162683949424445,0.747827661115134,42.61126708984375,0.759996894168802,0.7098240185650745,0.8178020386564919,41.85026550292969,0.7626676236249696
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ecd82469cb960c9ac4d7a041589441ffbde220cd2b4d5d13db88be8b9e5b462
3
  size 539666601
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fb7a64f61f366e6a68665ad8dc9ce6ef2969d2c1d541fac6d0287fe3afd5628
3
  size 539666601