Upload with huggingface_hub

README.md

@@ -46,7 +46,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 1511 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 506 with parameters:
 ```
 {'batch_size': 512, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -71,7 +71,7 @@ Parameters of the fit()-Method:
     },
     "scheduler": "WarmupLinear",
     "steps_per_epoch": null,
-    "warmup_steps": 150,
+    "warmup_steps": 50,
     "weight_decay": 0.01
 }
 ```

eval/binary_classification_evaluation_results.csv

@@ -1,22 +1,10 @@
 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
-0,250,0.997094640539037,0.7775063514709473,0.4290909090909091,0.3597560975609756,0.5315315315315315,0.5997836589813232,0.4015929798612823,0.9971873647771528,14.814654350280762,0.42990654205607476,0.44660194174757284,0.4144144144144144,18.121349334716797,0.39469511161848114,0.997094640539037,0.6670409440994263,0.4290909090909091,0.3597560975609756,0.5315315315315315,0.8946690559387207,0.4015929798612823,0.997094640539037,0.7775063514709473,0.4290909090909091,0.3597560975609756,0.5315315315315315,0.5997836589813232,0.40052047878878116
-0,500,0.9972800890152685,0.717904806137085,0.4888888888888889,0.6376811594202898,0.3963963963963964,0.7079300880432129,0.43683990431617215,0.9973109970946406,16.22292709350586,0.4912280701754387,0.7,0.3783783783783784,16.22292709350586,0.434646572707398,0.9972800890152685,0.7511184811592102,0.4888888888888889,0.6376811594202898,0.3963963963963964,0.7642902135848999,0.4368399043161722,0.9972800890152685,0.7179047465324402,0.4888888888888889,0.6376811594202898,0.3963963963963964,0.7079300880432129,0.4372689047451726
-0,750,0.9973419051740124,0.7321048378944397,0.54,0.6067415730337079,0.4864864864864865,0.693566083908081,0.4636571119455636,0.9974037213327563,15.98095417022705,0.5353535353535355,0.6091954022988506,0.4774774774774775,16.859485626220703,0.4616846282962872,0.9973419051740124,0.7319768667221069,0.54,0.6067415730337079,0.4864864864864865,0.7828586101531982,0.4636571119455635,0.9973419051740124,0.7321048378944397,0.54,0.6067415730337079,0.4864864864864865,0.6935661435127258,0.46318521147366304
-0,1000,0.9974964455708722,0.7233495116233826,0.5303867403314918,0.6857142857142857,0.43243243243243246,0.6975173354148865,0.49401017905762307,0.9975273536502441,16.29550552368164,0.5402843601895734,0.57,0.5135135135135135,17.95020866394043,0.49580376521780456,0.9974964455708722,0.7438418865203857,0.5303867403314918,0.6857142857142857,0.43243243243243246,0.7777924537658691,0.49401017905762307,0.9974964455708722,0.7233494520187378,0.5303867403314918,0.6857142857142857,0.43243243243243246,0.6975173950195312,0.49401017905762307
-0,1250,0.9974037213327563,0.7330388426780701,0.5472636815920398,0.6111111111111112,0.4954954954954955,0.694030225276947,0.48703844489790515,0.9974346294121283,15.799694061279297,0.5425531914893618,0.6623376623376623,0.4594594594594595,16.519229888916016,0.48615138963181054,0.9974037213327563,0.7306920289993286,0.5472636815920398,0.6111111111111112,0.4954954954954955,0.7822656631469727,0.4870384448979051,0.9974037213327563,0.7330388426780701,0.5472636815920398,0.6111111111111112,0.4954954954954955,0.6940302848815918,0.49476045261991286
-0,1500,0.9974346294121283,0.7500821948051453,0.5596330275229359,0.5700934579439252,0.5495495495495496,0.6585733890533447,0.5231257383043372,0.9974655374915002,15.052704811096191,0.5636363636363636,0.5688073394495413,0.5585585585585585,17.916898727416992,0.5243987833775768,0.9974346294121283,0.7069872617721558,0.5596330275229359,0.5700934579439252,0.5495495495495496,0.8263493776321411,0.5231257383043373,0.9974346294121283,0.7500821948051453,0.5596330275229359,0.5700934579439252,0.5495495495495496,0.6585733294487,0.522053237231836
-0,-1,0.9974964455708722,0.7362957000732422,0.5656565656565656,0.6436781609195402,0.5045045045045045,0.7011962532997131,0.5246028816802968,0.9974655374915002,14.561671257019043,0.5688073394495412,0.5794392523364486,0.5585585585585585,17.39862823486328,0.527904021164643,0.9974964455708722,0.7262285947799683,0.5656565656565656,0.6436781609195402,0.5045045045045045,0.7730506658554077,0.5246028816802968,0.9974964455708722,0.7362956404685974,0.5656565656565656,0.6436781609195402,0.5045045045045045,0.7011963129043579,0.5235303806077958
-1,250,0.99762007788836,0.7355296015739441,0.5760869565217391,0.726027397260274,0.4774774774774775,0.7128541469573975,0.5377334548451288,0.997681894047104,15.798933029174805,0.5775401069518716,0.7105263157894737,0.4864864864864865,16.66633415222168,0.5383354753231566,0.99762007788836,0.7272816300392151,0.5760869565217391,0.726027397260274,0.4774774774774775,0.7578203678131104,0.5377334548451288,0.99762007788836,0.7355296611785889,0.5760869565217391,0.726027397260274,0.4774774774774775,0.7128540873527527,0.5372615543732283
-1,500,0.9975273536502441,0.7189757823944092,0.5739910313901345,0.5714285714285714,0.5765765765765766,0.646925687789917,0.5097487388697948,0.9975582617296161,17.005178451538086,0.5791855203619908,0.5818181818181818,0.5765765765765766,18.30899429321289,0.5114833241403417,0.9975273536502441,0.7496980428695679,0.5739910313901345,0.5714285714285714,0.5765765765765766,0.8403255939483643,0.5097487388697947,0.9975273536502441,0.7189757823944092,0.5739910313901345,0.5714285714285714,0.5765765765765766,0.646925687789917,0.5056732347942906
-1,750,0.99762007788836,0.7455284595489502,0.582010582010582,0.7051282051282052,0.4954954954954955,0.6989697813987732,0.5291233316744711,0.9976509859677319,15.55929183959961,0.5789473684210527,0.6962025316455697,0.4954954954954955,16.8787841796875,0.5290634810360899,0.99762007788836,0.713379442691803,0.582010582010582,0.7051282051282052,0.4954954954954955,0.7759250402450562,0.5291233316744711,0.99762007788836,0.7455284595489502,0.582010582010582,0.7051282051282052,0.4954954954954955,0.6989697813987732,0.5291233316744711
-1,1000,0.9975273536502441,0.7243247032165527,0.5607476635514018,0.5825242718446602,0.5405405405405406,0.6744934320449829,0.5309383366409427,0.9975273536502441,15.974703788757324,0.5596330275229359,0.5700934579439252,0.5495495495495496,17.679428100585938,0.5301300700219603,0.9975273536502441,0.7425298690795898,0.5607476635514018,0.5825242718446602,0.5405405405405406,0.8068538308143616,0.5309383366409426,0.9975273536502441,0.7243247032165527,0.5607476635514018,0.5825242718446602,0.5405405405405406,0.6744934320449829,0.5306809363835424
-1,1250,0.99762007788836,0.7127195596694946,0.5759162303664921,0.6875,0.4954954954954955,0.6952959895133972,0.5315341138634572,0.99762007788836,16.535133361816406,0.5760869565217391,0.726027397260274,0.4774774774774775,16.757211685180664,0.5342964968834436,0.99762007788836,0.7579857110977173,0.5759162303664921,0.6875,0.4954954954954955,0.7806457281112671,0.5315341138634572,0.99762007788836,0.7127195000648499,0.5759162303664921,0.6875,0.4954954954954955,0.695296049118042,0.533464615793959
-1,1500,0.9974964455708722,0.7703023552894592,0.5803108808290156,0.6829268292682927,0.5045045045045045,0.7004988193511963,0.5293702966413293,0.9974964455708722,16.719430923461914,0.5888324873096448,0.6744186046511628,0.5225225225225225,17.071949005126953,0.531370734009549,0.9974964455708722,0.6777868270874023,0.5803108808290156,0.6829268292682927,0.5045045045045045,0.7739505767822266,0.5293702966413293,0.9974964455708722,0.770302414894104,0.5803108808290156,0.6829268292682927,0.5045045045045045,0.7004988193511963,0.5345183017893343
-1,-1,0.9974655374915002,0.7704312801361084,0.5829145728643216,0.6590909090909091,0.5225225225225225,0.692418098449707,0.5294366403102271,0.9974655374915002,14.78571891784668,0.5841584158415841,0.6483516483516484,0.5315315315315315,17.190946578979492,0.5314405448295747,0.9974655374915002,0.6775965690612793,0.5829145728643216,0.6590909090909091,0.5225225225225225,0.7843228578567505,0.529436640310227,0.9974655374915002,0.7704312801361084,0.5829145728643216,0.6590909090909091,0.5225225225225225,0.6924181580543518,0.5253611362347229
-2,250,0.997681894047104,0.7062087655067444,0.5989304812834224,0.7368421052631579,0.5045045045045045,0.7062087655067444,0.5352920944901863,0.9977128021264758,16.72159194946289,0.6122448979591837,0.7058823529411765,0.5405405405405406,17.02515411376953,0.5372647651944455,0.997681894047104,0.7665382623672485,0.5989304812834224,0.7368421052631579,0.5045045045045045,0.7665382623672485,0.5352920944901864,0.997681894047104,0.7062088251113892,0.5989304812834224,0.7368421052631579,0.5045045045045045,0.7062088251113892,0.5372227474761445
-2,500,0.997681894047104,0.7194023132324219,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6969935297966003,0.5378354767854148,0.9977128021264758,16.674388885498047,0.6063829787234042,0.7402597402597403,0.5135135135135135,16.674388885498047,0.5434424568138472,0.997681894047104,0.7491283416748047,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.7784652709960938,0.5378354767854149,0.997681894047104,0.7194023132324219,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6969934701919556,0.5367629757129138
-2,750,0.997681894047104,0.7215240001678467,0.5925925925925926,0.717948717948718,0.5045045045045045,0.7041295766830444,0.5389989304663247,0.9977128021264758,16.45661163330078,0.6051282051282051,0.7023809523809523,0.5315315315315315,17.144792556762695,0.5413865910302602,0.997681894047104,0.7462912201881409,0.5925925925925926,0.717948717948718,0.5045045045045045,0.7692457437515259,0.5389989304663249,0.997681894047104,0.7215239405632019,0.5925925925925926,0.717948717948718,0.5045045045045045,0.7041295766830444,0.5409294323968268
-2,1000,0.9977128021264758,0.7257504463195801,0.5934065934065934,0.7605633802816901,0.4864864864864865,0.721716046333313,0.545513069356616,0.9977128021264758,15.917673110961914,0.5934065934065934,0.7605633802816901,0.4864864864864865,16.44021224975586,0.5465690431234398,0.9977128021264758,0.7406044602394104,0.5934065934065934,0.7605633802816901,0.4864864864864865,0.7460346221923828,0.5455130693566161,0.9977128021264758,0.7257504463195801,0.5934065934065934,0.7605633802816901,0.4864864864864865,0.7217161059379578,0.5443547681983151
-2,1250,0.9977128021264758,0.707938551902771,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6963058710098267,0.5468800586528105,0.9977128021264758,16.721351623535156,0.6113989637305699,0.7195121951219512,0.5315315315315315,16.945556640625,0.5489303554702061,0.9977128021264758,0.7642788887023926,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.7793498635292053,0.5468800586528105,0.9977128021264758,0.7079386711120605,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6963059306144714,0.542418454191206
-2,1500,0.9977128021264758,0.7062461376190186,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6944886445999146,0.5483262325355656,0.9977437102058478,16.841472625732422,0.6145833333333334,0.7283950617283951,0.5315315315315315,16.965425491333008,0.5496214256679012,0.9977128021264758,0.7664884924888611,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.7816791534423828,0.5483262325355656,0.9977128021264758,0.7062461376190186,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6944886445999146,0.5483262325355656
-2,-1,0.9977128021264758,0.7062256336212158,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6945241689682007,0.5483685187984644,0.9977437102058478,16.842079162597656,0.6145833333333334,0.7283950617283951,0.5315315315315315,16.965831756591797,0.549552325857223,0.9977128021264758,0.7665152549743652,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.781633734703064,0.5483685187984643,0.9977128021264758,0.7062256336212158,0.6041666666666666,0.7160493827160493,0.5225225225225225,0.6945241689682007,0.5515002219301675
+0,250,0.997063732459665,0.7604681253433228,0.4975609756097561,0.5425531914893617,0.4594594594594595,0.6962460875511169,0.4108009109682283,0.997094640539037,14.697938919067383,0.5,0.5142857142857142,0.4864864864864865,17.019569396972656,0.40722923692362134,0.997063732459665,0.6921424865722656,0.4975609756097561,0.5425531914893617,0.4594594594594595,0.7794278264045715,0.4108009109682283,0.997063732459665,0.7604681253433228,0.4975609756097561,0.5425531914893617,0.4594594594594595,0.6962461471557617,0.41594891611623347
+0,500,0.9972182728565246,0.7466210126876831,0.48401826484018273,0.49074074074074076,0.4774774774774775,0.6766296625137329,0.4456477145925501,0.9972491809358967,15.441628456115723,0.4811320754716982,0.504950495049505,0.4594594594594595,17.22469711303711,0.4429553568312313,0.9972182728565246,0.7118691802024841,0.48401826484018273,0.49074074074074076,0.4774774774774775,0.8042017817497253,0.44564768978597324,0.9972182728565246,0.7466210126876831,0.48401826484018273,0.49074074074074076,0.4774774774774775,0.6766296625137329,0.4533697223145578
+0,-1,0.9972491809358967,0.7341707944869995,0.4821428571428572,0.4778761061946903,0.4864864864864865,0.6649925112724304,0.447053933023235,0.9973109970946406,15.690232276916504,0.4782608695652174,0.46218487394957986,0.4954954954954955,17.765544891357422,0.44314733088854175,0.9972491809358967,0.7291487455368042,0.4821428571428572,0.4778761061946903,0.4864864864864865,0.8185443878173828,0.4470539330232351,0.9972491809358967,0.7341707944869995,0.4821428571428572,0.4778761061946903,0.4864864864864865,0.6649925112724304,0.45348893945824137
+1,250,0.9972182728565246,0.7542093992233276,0.4651162790697675,0.4807692307692308,0.45045045045045046,0.6726086139678955,0.4203797743032279,0.9972182728565246,15.242410659790039,0.4549763033175356,0.48,0.43243243243243246,17.406028747558594,0.417422584255281,0.9972182728565246,0.701128363609314,0.4651162790697675,0.4807692307692308,0.45045045045045046,0.8091862201690674,0.4203797743032279,0.9972182728565246,0.7542095184326172,0.4651162790697675,0.4807692307692308,0.45045045045045046,0.6726086139678955,0.42810178202523563
+1,500,0.9974346294121283,0.7397375106811523,0.5120772946859903,0.5520833333333334,0.4774774774774775,0.6753202676773071,0.48274800818963626,0.9974346294121283,15.669346809387207,0.5071770334928231,0.5408163265306123,0.4774774774774775,17.518417358398438,0.4813764003960228,0.9974346294121283,0.7214730978012085,0.5120772946859903,0.5520833333333334,0.4774774774774775,0.8058284521102905,0.4827480081896363,0.9974346294121283,0.7397375106811523,0.5120772946859903,0.5520833333333334,0.4774774774774775,0.6753202676773071,0.48047430591593393
+1,-1,0.9974346294121283,0.7395174503326416,0.5125628140703518,0.5795454545454546,0.4594594594594595,0.6868640184402466,0.483239902840429,0.9974346294121283,15.812583923339844,0.5025125628140704,0.5681818181818182,0.45045045045045046,17.09825897216797,0.4824843649168539,0.9974346294121283,0.7217788696289062,0.5125628140703518,0.5795454545454546,0.4594594594594595,0.7913718223571777,0.48323990284042906,0.9974346294121283,0.7395174503326416,0.5125628140703518,0.5795454545454546,0.4594594594594595,0.6868640184402466,0.479164398764925
+2,250,0.9974346294121283,0.7224153876304626,0.5320197044334976,0.5869565217391305,0.4864864864864865,0.667343020439148,0.4753145499816797,0.9974655374915002,16.396638870239258,0.5352112676056339,0.5588235294117647,0.5135135135135135,17.98391342163086,0.47710728495811605,0.9974346294121283,0.7450753450393677,0.5320197044334976,0.5869565217391305,0.4864864864864865,0.8156678080558777,0.4753145499816797,0.9974346294121283,0.7224153280258179,0.5320197044334976,0.5869565217391305,0.4864864864864865,0.6673429608345032,0.47085294552007534
+2,500,0.9974964455708722,0.724300742149353,0.5308056872037915,0.56,0.5045045045045045,0.6692973971366882,0.48360038042538306,0.9975273536502441,16.06214141845703,0.5258215962441315,0.5490196078431373,0.5045045045045045,17.813426971435547,0.48376237836955577,0.9974964455708722,0.7425557374954224,0.5308056872037915,0.56,0.5045045045045045,0.8132681846618652,0.4836003804253831,0.9974964455708722,0.7243008613586426,0.5308056872037915,0.56,0.5045045045045045,0.6692973971366882,0.488533885358888
+2,-1,0.9974964455708722,0.7243217825889587,0.5308056872037915,0.56,0.5045045045045045,0.6693127155303955,0.4835313137096024,0.9975273536502441,16.063438415527344,0.5258215962441315,0.5490196078431373,0.5045045045045045,17.81356430053711,0.48375084785483957,0.9974964455708722,0.742527425289154,0.5308056872037915,0.56,0.5045045045045045,0.8132493495941162,0.4835313137096024,0.9974964455708722,0.724321722984314,0.5308056872037915,0.56,0.5045045045045045,0.6693127155303955,0.49125332143161016

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ce28ed698e37cbbea369622784abb346edcd487a93e1ea3a26cc329e6154547e
+oid sha256:2fe9eb21b439a895f395de6d0ee0a25d3824ced45eaa42202b1bb55ec44bf585
 size 438014769