,avg_role_model_loss_train,avg_role_model_std_loss_train,avg_role_model_mean_pred_loss_train,avg_role_model_g_mag_loss_train,avg_role_model_g_cos_loss_train,avg_non_role_model_g_mag_loss_train,avg_non_role_model_g_cos_loss_train,avg_non_role_model_embed_loss_train,avg_loss_train,n_size_train,n_batch_train,duration_train,duration_batch_train,duration_size_train,avg_pred_std_train,avg_role_model_loss_test,avg_role_model_std_loss_test,avg_role_model_mean_pred_loss_test,avg_role_model_g_mag_loss_test,avg_role_model_g_cos_loss_test,avg_non_role_model_g_mag_loss_test,avg_non_role_model_g_cos_loss_test,avg_non_role_model_embed_loss_test,avg_loss_test,n_size_test,n_batch_test,duration_test,duration_batch_test,duration_size_test,avg_pred_std_test
0,0.07503844532329822,3.76196769104788,0.039352887886764186,0.0,0.0,0.0,0.0,0.0,0.07503844532329822,320,40,42.94257736206055,1.0735644340515136,0.1341955542564392,0.13469835626892745,0.008877724732155912,3.33818835544007,2.9330407841143823e-06,0.0,0.0,0.0,0.0,0.0,0.008877724732155912,80,10,9.165157794952393,0.9165157794952392,0.1145644724369049,0.037592777702957395
1,0.02608004305366194,2.0130314562969813,0.005282479949307373,0.0,0.0,0.0,0.0,0.0,0.02608004305366194,320,40,42.331037521362305,1.0582759380340576,0.1322844922542572,0.10117223353590817,0.005078719957964495,0.2115427433644072,3.133896269957859e-05,0.0,0.0,0.0,0.0,0.0,0.005078719957964495,80,10,9.217864990234375,0.9217864990234375,0.11522331237792968,0.11006196644157171
2,0.007246867158391979,3.0597032676599154,9.03313408463391e-05,0.0,0.0,0.0,0.0,0.0,0.007246867158391979,320,40,42.494264125823975,1.0623566031455993,0.13279457539319992,0.07669011817779392,0.0074968072643969205,2.55860044410183,0.00017504165297967944,0.0,0.0,0.0,0.0,0.0,0.0074968072643969205,80,10,9.166083335876465,0.9166083335876465,0.11457604169845581,0.09575922545045615
3,0.006846483054687269,2.2157006146561513,0.0007793176604440372,0.0,0.0,0.0,0.0,0.0,0.006846483054687269,320,40,42.40406584739685,1.0601016461849213,0.13251270577311516,0.08761264618951828,0.0020610511739505453,0.43892243231239264,7.913704455120296e-06,0.0,0.0,0.0,0.0,0.0,0.0020610511739505453,80,10,9.511794328689575,0.9511794328689576,0.1188974291086197,0.08304880987852811
4,0.002302334751948365,0.6305191363795544,1.4314678949647008e-05,0.0,0.0,0.0,0.0,0.0,0.002302334751948365,320,40,43.27270483970642,1.0818176209926604,0.13522720262408255,0.08970329142175615,0.004081522431806661,0.026551660033874214,2.7209868290256623e-05,0.0,0.0,0.0,0.0,0.0,0.004081522431806661,80,10,9.23831558227539,0.923831558227539,0.11547894477844238,0.11656681830063462
5,0.0013704985673030023,0.32384693517769847,7.155363357548236e-06,0.0,0.0,0.0,0.0,0.0,0.0013704985673030023,320,40,42.453025341033936,1.0613256335258483,0.13266570419073104,0.09745097612030804,0.0033582281379494817,0.619899958840142,2.357043052825247e-06,0.0,0.0,0.0,0.0,0.0,0.0033582281379494817,80,10,9.264163732528687,0.9264163732528686,0.11580204665660858,0.06247350247576833
6,0.0025572317323167225,1.3281221274127346,4.255108958927875e-06,0.0,0.0,0.0,0.0,0.0,0.0025572317323167225,320,40,42.2554829120636,1.05638707280159,0.13204838410019876,0.08552498414646834,0.001253202352381777,0.32618888739889373,2.2838767717248133e-06,0.0,0.0,0.0,0.0,0.0,0.001253202352381777,80,10,9.205610513687134,0.9205610513687134,0.11507013142108917,0.07330623050220311
7,0.0017415513455489417,0.39128143024250334,1.98570894781383e-06,0.0,0.0,0.0,0.0,0.0,0.0017415513455489417,320,40,42.568729639053345,1.0642182409763337,0.1330272801220417,0.08582097220933065,0.002552773474599235,0.38444976235623474,9.21505393498695e-06,0.0,0.0,0.0,0.0,0.0,0.002552773474599235,80,10,9.204639673233032,0.9204639673233033,0.11505799591541291,0.09078566757962107
8,0.001197526408395788,0.5185240543789404,6.638705742609621e-06,0.0,0.0,0.0,0.0,0.0,0.001197526408395788,320,40,42.46436643600464,1.061609160900116,0.1327011451125145,0.0921793600777164,0.0011125051009003074,0.1445327332803572,3.7028677351003125e-06,0.0,0.0,0.0,0.0,0.0,0.0011125051009003074,80,10,9.251813411712646,0.9251813411712646,0.11564766764640808,0.08213724349625409
9,0.0011423767211454106,0.24242009912380066,5.1029623472642616e-06,0.0,0.0,0.0,0.0,0.0,0.0011423767211454106,320,40,42.30474233627319,1.0576185584068298,0.13220231980085373,0.086794763058424,0.0021649388814694247,1.9614427807347965,1.2043508045372908e-05,0.0,0.0,0.0,0.0,0.0,0.0021649388814694247,80,10,9.148065567016602,0.9148065567016601,0.11435081958770751,0.0991100890096277
10,0.0008830112970827031,0.4022787155074184,5.028790277500362e-07,0.0,0.0,0.0,0.0,0.0,0.0008830112970827031,320,40,42.54185461997986,1.0635463654994965,0.13294329568743707,0.09175913570215925,0.0017155635854578578,1.2685116354904722,2.10015661070706e-06,0.0,0.0,0.0,0.0,0.0,0.0017155635854578578,80,10,9.253859281539917,0.9253859281539917,0.11567324101924896,0.0978053328813985
11,0.001937569323945354,0.6675240401481404,3.123376906712105e-06,0.0,0.0,0.0,0.0,0.0,0.001937569323945354,320,40,42.27454662322998,1.0568636655807495,0.1321079581975937,0.09221202009357513,0.0027879032801138236,0.09599128968548029,1.215036173931594e-05,0.0,0.0,0.0,0.0,0.0,0.0027879032801138236,80,10,9.17811369895935,0.917811369895935,0.11472642123699188,0.11065587596967816
12,0.0013319606783625203,0.16566794978843974,6.807524444540914e-07,0.0,0.0,0.0,0.0,0.0,0.0013319606783625203,320,40,42.627525091171265,1.0656881272792815,0.1332110159099102,0.10023473438341171,0.0013272355950903147,0.5792492911004956,6.966086060211652e-08,0.0,0.0,0.0,0.0,0.0,0.0013272355950903147,80,10,9.216788053512573,0.9216788053512573,0.11520985066890717,0.07118493653833866
13,0.0007021169698418816,0.11813758928258485,2.759127278142287e-06,0.0,0.0,0.0,0.0,0.0,0.0007021169698418816,320,40,42.255826234817505,1.0563956558704377,0.1320494569838047,0.09224181645549834,0.0008028386626392602,0.06161341504857774,6.375153506842091e-07,0.0,0.0,0.0,0.0,0.0,0.0008028386626392602,80,10,9.26439380645752,0.9264393806457519,0.11580492258071899,0.0892744664568454
14,0.0006613305880819098,0.18104081245551243,6.271647721827747e-07,0.0,0.0,0.0,0.0,0.0,0.0006613305880819098,320,40,42.603567361831665,1.0650891840457917,0.13313614800572396,0.09589193011634052,0.0005620345647912473,0.01514620759198806,4.4228354818542924e-07,0.0,0.0,0.0,0.0,0.0,0.0005620345647912473,80,10,9.197651863098145,0.9197651863098144,0.1149706482887268,0.09222434270195663
15,0.0007842243476261501,0.3221512252403457,1.0985442627384213e-06,0.0,0.0,0.0,0.0,0.0,0.0007842243476261501,320,40,42.46329689025879,1.0615824222564698,0.13269780278205873,0.09709920620080084,0.001301504473667592,2.9538385085063057,4.959147403504893e-07,0.0,0.0,0.0,0.0,0.0,0.001301504473667592,80,10,9.24899411201477,0.924899411201477,0.11561242640018463,0.1033931726939045
16,0.013570401900506113,0.36384937064023576,0.0018209778673778428,0.0,0.0,0.0,0.0,0.0,0.013570401900506113,320,40,42.59553337097168,1.064888334274292,0.1331110417842865,0.12845125668682159,0.16798840463161469,0.3928926819935441,0.06907801991328597,0.0,0.0,0.0,0.0,0.0,0.16798840463161469,80,10,9.341678619384766,0.9341678619384766,0.11677098274230957,0.28858067095279694
17,0.22693241573870182,0.459286569285905,0.10300876491237432,0.0,0.0,0.0,0.0,0.0,0.22693241573870182,320,40,42.3380069732666,1.058450174331665,0.13230627179145812,0.32175534069538114,0.4309734970331192,0.6810152728110552,0.33519675582647324,0.0,0.0,0.0,0.0,0.0,0.4309734970331192,80,10,9.211932897567749,0.921193289756775,0.11514916121959687,0.35243880599737165
18,0.2022466917289421,0.7070020012586611,0.11807100524520138,0.0,0.0,0.0,0.0,0.0,0.2022466917289421,320,40,42.41661763191223,1.0604154407978057,0.1325519300997257,0.34771558828651905,0.023606129095423967,0.21133080043364316,0.0015780384962681636,0.0,0.0,0.0,0.0,0.0,0.023606129095423967,80,10,9.170459747314453,0.9170459747314453,0.11463074684143067,0.16341671436093747
19,0.005479642776481342,0.5328093900557633,8.112759967222604e-05,0.0,0.0,0.0,0.0,0.0,0.005479642776481342,320,40,42.17777228355408,1.0544443070888518,0.13180553838610648,0.1161046927794814,0.0007801399522577412,0.05935813882520051,5.190229413365443e-07,0.0,0.0,0.0,0.0,0.0,0.0007801399522577412,80,10,9.320383310317993,0.9320383310317993,0.11650479137897492,0.0829056172631681
20,0.0016211183414270637,0.5920635455369594,2.3483921812016834e-06,0.0,0.0,0.0,0.0,0.0,0.0016211183414270637,320,40,42.64041256904602,1.0660103142261506,0.13325128927826882,0.08777563656913116,0.0008631729724584147,0.09347999086021445,1.679538957422011e-06,0.0,0.0,0.0,0.0,0.0,0.0008631729724584147,80,10,9.24825143814087,0.9248251438140869,0.11560314297676086,0.07839330667629837
21,0.0005253879406154737,0.10636353976760801,2.6855408757735233e-07,0.0,0.0,0.0,0.0,0.0,0.0005253879406154737,320,40,42.83327293395996,1.070831823348999,0.13385397791862488,0.09573199808364734,0.0005467957133078016,0.23533951704739592,2.0549961012861218e-07,0.0,0.0,0.0,0.0,0.0,0.0005467957133078016,80,10,9.286863803863525,0.9286863803863525,0.11608579754829407,0.08401567195542156
22,0.0006425162649065896,0.10582681066216537,6.702316121443182e-07,0.0,0.0,0.0,0.0,0.0,0.0006425162649065896,320,40,42.88295650482178,1.0720739126205445,0.13400923907756807,0.09285907801240682,0.003129586172872223,0.14476592368632737,2.8466294405604663e-05,0.0,0.0,0.0,0.0,0.0,0.003129586172872223,80,10,9.417258024215698,0.9417258024215698,0.11771572530269622,0.10630810875445604
23,0.0012267698623873002,0.3422558290479515,5.092052370217958e-06,0.0,0.0,0.0,0.0,0.0,0.0012267698623873002,320,40,42.59943246841431,1.0649858117103577,0.1331232264637947,0.0916012367233634,0.001950129849865334,1.6070946810563327,1.2394382595015685e-05,0.0,0.0,0.0,0.0,0.0,0.001950129849865334,80,10,9.218074083328247,0.9218074083328247,0.11522592604160309,0.06952399904839694
24,0.0015580077400954907,0.5951609104130398,1.101507371748138e-05,0.0,0.0,0.0,0.0,0.0,0.0015580077400954907,320,40,42.20673155784607,1.0551682889461518,0.13189603611826897,0.08924343169201165,0.0014282745076343417,1.1416928244575502,5.442704249958296e-07,0.0,0.0,0.0,0.0,0.0,0.0014282745076343417,80,10,9.161205768585205,0.9161205768585206,0.11451507210731507,0.08108144407160581
25,0.001158999443759967,0.673237547548581,1.6942943679773905e-06,0.0,0.0,0.0,0.0,0.0,0.001158999443759967,320,40,42.81976580619812,1.070494145154953,0.13381176814436913,0.09180414919974282,0.003335691889151349,0.1234515317961268,9.410348545632954e-05,0.0,0.0,0.0,0.0,0.0,0.003335691889151349,80,10,9.541585206985474,0.9541585206985473,0.11926981508731842,0.10400733416900039