| ,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.01341511887658271,0.24023109539175563,0.000575269428866676,0.004837815280916402,0.0,0.0,0.0,0.0,0.015502669336274266,320,80,86.60318899154663,1.082539862394333,0.27063496559858324,0.11721891383640468,0.018764166883192955,0.7378902015632776,0.0008810971392904321,0.006337779993191362,0.0,0.0,0.0,0.0,0.02165449762251228,80,20,19.204163312911987,0.9602081656455994,0.24005204141139985,0.08749947492033243 | |
| 1,0.01306824244238669,1.023421409522797,0.0006003775347632523,0.007479726555175148,0.0,0.0,0.0,0.0,0.014787302106560674,320,80,87.21888208389282,1.0902360260486603,0.2725590065121651,0.1071490949485451,0.007755115552572534,2.593804732917124,0.00013148071556594586,0.006523144242237322,0.0,0.0,0.0,0.0,0.008876382285961881,80,20,19.415016651153564,0.9707508325576782,0.24268770813941956,0.026741976058110593 | |
| 2,0.0028183516405988485,0.7737382953319809,9.625664277708966e-06,0.00837607061257586,0.0,0.0,0.0,0.0,0.0028855871143605325,320,80,87.30349731445312,1.091293716430664,0.272823429107666,0.0851033627637662,0.0034189124999102205,2.2048701629042626,1.1537741464451301e-05,0.008503273967653513,0.0,0.0,0.0,0.0,0.0035019157105125485,80,20,19.296178340911865,0.9648089170455932,0.2412022292613983,0.02689541974104941 | |
| 3,0.0031308751182223204,1.4198307410230258,1.015549445890062e-05,0.008551960467593744,0.0,0.0,0.0,0.0,0.0032078142976388335,320,80,87.20788407325745,1.0900985509157182,0.27252463772892954,0.07063841636409052,0.0025563906941897586,1.7120467301925602,4.022160688882392e-06,0.008696647034958005,0.0,0.0,0.0,0.0,0.0026055111757159466,80,20,19.3008930683136,0.9650446534156799,0.24126116335391998,0.031063590943813325 | |
| 4,0.0020325787663750816,0.9927601797127522,3.947625345419869e-06,0.008560203079832717,0.0,0.0,0.0,0.0,0.002069109250442125,320,80,86.83163189888,1.085395398736,0.271348849684,0.08267453832668252,0.002587378228054149,1.5380892607378684,5.001206458744855e-06,0.008236717758700251,0.0,0.0,0.0,0.0,0.002642228409240488,80,20,19.28103756904602,0.964051878452301,0.24101296961307525,0.03489238116890192 | |
| 5,0.0018827232423063833,0.36071941533007446,3.0995558925314963e-06,0.008657984499586746,0.0,0.0,0.0,0.0,0.0019155390065861866,320,80,86.8512351512909,1.085640439391136,0.271410109847784,0.08422506358474494,0.002631575356645044,1.174698173921024,7.702732040915094e-06,0.008593563642352819,0.0,0.0,0.0,0.0,0.002682924266991904,80,20,19.29587149620056,0.964793574810028,0.241198393702507,0.03935151700861752 | |
| 6,0.00163584444890148,0.32704864908948517,3.7024563248498977e-06,0.008764190669171511,0.0,0.0,0.0,0.0,0.0016616366719972575,320,80,86.82903480529785,1.085362935066223,0.27134073376655576,0.09021815697196871,0.0027584389958065004,1.2948427033639747,8.096104292887318e-06,0.007971254212316125,0.0,0.0,0.0,0.0,0.002821568233775906,80,20,19.269461393356323,0.9634730696678162,0.24086826741695405,0.03149411482736468 | |
| 7,0.001588404598624038,0.30104452142173416,2.1237631645752536e-06,0.008680318144615739,0.0,0.0,0.0,0.0,0.0016122827923027217,320,80,86.72441625595093,1.0840552031993866,0.27101380079984666,0.08948315244633705,0.002168826656998135,1.1961765073471269,3.218722386769124e-06,0.008597346721217036,0.0,0.0,0.0,0.0,0.0022069613172789105,80,20,19.133681535720825,0.9566840767860413,0.23917101919651032,0.03733009579591453 | |
| 8,0.001353109241972561,0.2765715855718042,2.087236364796191e-06,0.008927629049867391,0.0,0.0,0.0,0.0,0.0013720194954657927,320,80,86.93680453300476,1.0867100566625596,0.2716775141656399,0.09051385981729254,0.0024928924576670397,0.8476081335626077,6.1137728412461815e-06,0.008560518850572407,0.0,0.0,0.0,0.0,0.0025436472140427215,80,20,19.220860481262207,0.9610430240631104,0.2402607560157776,0.03947796570137143 | |
| 9,0.0011078594061473268,0.2964725857345002,1.1485152926839908e-06,0.009000935748917981,0.0,0.0,0.0,0.0,0.0011219609619729453,320,80,86.97083497047424,1.087135437130928,0.271783859282732,0.08712627965724096,0.002337238602922298,0.6530551572132708,6.45294174761446e-06,0.00837808190844953,0.0,0.0,0.0,0.0,0.0023841866728616878,80,20,19.421545267105103,0.9710772633552551,0.24276931583881378,0.04457564675249159 | |
| 10,0.0012053420616211952,0.1931557533955261,9.706161082351402e-07,0.008857980108587071,0.0,0.0,0.0,0.0,0.001221423167589819,320,80,87.4618911743164,1.093273639678955,0.27331840991973877,0.0897115994244814,0.002253816397569608,0.8841316987068751,4.704099107730175e-06,0.008425972750410437,0.0,0.0,0.0,0.0,0.0022973355487920346,80,20,19.446954488754272,0.9723477244377137,0.24308693110942842,0.04312506481073797 | |
| 11,0.0009794028399483067,0.1359832607921138,1.01823794474289e-06,0.009016435348894448,0.0,0.0,0.0,0.0,0.0009917656654579333,320,80,87.36561322212219,1.0920701652765274,0.27301754131913186,0.08817402567947283,0.0025446744170039893,0.6562726007028459,8.59153058119233e-06,0.00828958151396364,0.0,0.0,0.0,0.0,0.002601461045560427,80,20,19.171342372894287,0.9585671186447143,0.23964177966117858,0.046762616652995345 | |
| 12,0.0008729144968128821,0.1683987133885907,6.835936173131307e-07,0.009030447795521469,0.0,0.0,0.0,0.0,0.0008834377124003368,320,80,87.13745355606079,1.0892181694507599,0.27230454236268997,0.09424625603714958,0.0025075858677155336,0.6813347520466777,9.080454932988875e-06,0.007943603885360062,0.0,0.0,0.0,0.0,0.002564889276982285,80,20,19.267545700073242,0.9633772850036622,0.24084432125091554,0.04623653790913522 | |
| 13,0.0008042755165661219,0.2813704043714537,3.3645215451869243e-07,0.009021570929326117,0.0,0.0,0.0,0.0,0.0008135768726788229,320,80,86.61275219917297,1.082659402489662,0.2706648506224155,0.08749473023926839,0.0024872228954336607,0.8424709204863575,1.2303503870288169e-05,0.00836391884367913,0.0,0.0,0.0,0.0,0.00254274607723346,80,20,19.089064598083496,0.9544532299041748,0.2386133074760437,0.04271733276546001 | |
| 14,0.0008596906129241689,0.12678315311761706,7.324179896302241e-07,0.009051537868799642,0.0,0.0,0.0,0.0,0.0008700103617229615,320,80,86.11309504508972,1.0764136880636215,0.26910342201590537,0.09114952590316534,0.003142218668654095,0.5989244450749837,1.701286636457411e-05,0.007768401200883091,0.0,0.0,0.0,0.0,0.0032341236757929435,80,20,18.97400975227356,0.948700487613678,0.2371751219034195,0.045658442331478 | |
| 15,0.0007071739214552508,0.11358937955730664,8.504173889859433e-07,0.009156818711198866,0.0,0.0,0.0,0.0,0.0007155208293625037,320,80,85.8247458934784,1.07280932366848,0.26820233091712,0.08828951774630696,0.0027607452910160648,0.5599186637941784,1.3696136823915239e-05,0.00771399496588856,0.0,0.0,0.0,0.0,0.0028348844876745715,80,20,18.95707416534424,0.9478537082672119,0.23696342706680298,0.04666830957867205 | |
| 16,0.0006871584620967042,0.056319261586604344,7.729947919388539e-07,0.009021204826422036,0.0,0.0,0.0,0.0,0.0006950973430775775,320,80,85.77924585342407,1.0722405731678009,0.2680601432919502,0.09376501713413746,0.0029137152865587267,0.6469025506016806,9.816658314748538e-06,0.007721508503891528,0.0,0.0,0.0,0.0,0.002987763029523194,80,20,18.965726137161255,0.9482863068580627,0.23707157671451567,0.048026554053649306 | |
| 17,0.0006477722853333034,0.08796985171551344,5.949152889407616e-07,0.009104125620797276,0.0,0.0,0.0,0.0,0.0006556250523999552,320,80,86.42096662521362,1.0802620828151703,0.2700655207037926,0.09506104957545176,0.003072608428192325,0.6891105846539176,1.3079660551437028e-05,0.007593843806535006,0.0,0.0,0.0,0.0,0.0031587239500368014,80,20,19.130789279937744,0.9565394639968872,0.2391348659992218,0.04223878695629537 | |
| 18,0.0006514384161164343,0.09940106079761221,1.795125833634764e-07,0.009067119332030416,0.0,0.0,0.0,0.0,0.0006588086861484044,320,80,87.44042205810547,1.0930052757263184,0.2732513189315796,0.09078157264739276,0.0024777524726232515,0.7228478452472927,1.1093569846298834e-05,0.008034943602979183,0.0,0.0,0.0,0.0,0.002537216582277324,80,20,19.3044171333313,0.965220856666565,0.24130521416664125,0.04896460571326315 | |
| 19,0.0006081814904973726,0.09099644586050033,3.1757028354558596e-07,0.009112988878041507,0.0,0.0,0.0,0.0,0.000615051853355908,320,80,86.71533060073853,1.0839416325092315,0.27098540812730787,0.09587977258488536,0.00240422225324437,0.6358416218907224,1.1121995321849986e-05,0.008140259771607816,0.0,0.0,0.0,0.0,0.0024615431102574803,80,20,19.22822141647339,0.9614110708236694,0.24035276770591735,0.04650567690841854 | |
| 20,0.0005358590460900814,0.06825528511013328,2.678394117848915e-07,0.009244002948980779,0.0,0.0,0.0,0.0,0.0005420027183504316,320,80,85.95558786392212,1.0744448482990265,0.26861121207475663,0.0906242580153048,0.0025660589744802564,0.6802260238559029,9.92956469314521e-06,0.007699522981420159,0.0,0.0,0.0,0.0,0.0026287092332495376,80,20,18.98186469078064,0.9490932345390319,0.23727330863475798,0.04403721652925015 | |
| 21,0.0004775314865582914,0.06455386250157247,2.1060846659059965e-07,0.009264782385434956,0.0,0.0,0.0,0.0,0.00048304505050964506,320,80,85.94736814498901,1.0743421018123627,0.26858552545309067,0.09594059572555125,0.0030469499950413594,0.8013474333885384,1.4471279196381915e-05,0.007496322155930102,0.0,0.0,0.0,0.0,0.0031343029033450874,80,20,19.01077938079834,0.950538969039917,0.23763474225997924,0.04275354435667396 | |
| 22,0.0005181060546192384,0.049453351887659557,3.420242802625219e-07,0.009284257958643138,0.0,0.0,0.0,0.0,0.0005241578612640296,320,80,85.98614287376404,1.0748267859220504,0.2687066964805126,0.09294412531889976,0.002864245133969234,0.4667397131830512,1.4686058315405149e-05,0.007691610814072191,0.0,0.0,0.0,0.0,0.002942165515560191,80,20,19.0608127117157,0.9530406355857849,0.23826015889644622,0.052058699540793896 | |
| 23,0.00043534455560347853,0.07720435771696674,3.792708702435137e-07,0.009331031411420554,0.0,0.0,0.0,0.0,0.00044047548610706146,320,80,85.86611938476562,1.0733264923095702,0.26833162307739256,0.09743042136542499,0.0030215614737244324,0.6433068920836377,1.4186154598316848e-05,0.007460616645403206,0.0,0.0,0.0,0.0,0.003107524223742075,80,20,18.98639988899231,0.9493199944496155,0.23732999861240386,0.050000256625935435 | |
| 24,0.00042359301896794933,0.04604176523210342,1.3871310008198302e-07,0.009280809713527561,0.0,0.0,0.0,0.0,0.000428745166027511,320,80,86.06277060508728,1.075784632563591,0.26894615814089773,0.09038203665986658,0.0026232202184473864,0.575840337219779,1.1411215524770312e-05,0.007816769555211068,0.0,0.0,0.0,0.0,0.002690732198243495,80,20,19.039226293563843,0.9519613146781921,0.23799032866954803,0.04816291746683419 | |
| 25,0.00037949440795728153,0.024586391349637894,1.68624920803169e-07,0.009385608148295432,0.0,0.0,0.0,0.0,0.00038402225013669524,320,80,86.02636766433716,1.0753295958042144,0.2688323989510536,0.09288427671417594,0.0026626741448126266,0.6883730123883651,1.143276741413235e-05,0.0078037152998149395,0.0,0.0,0.0,0.0,0.0027320246357703582,80,20,19.006802558898926,0.9503401279449463,0.23758503198623657,0.04854826694354415 | |
| 26,0.00036825626919494424,0.06836344334289635,2.864049477936933e-07,0.009369707445148378,0.0,0.0,0.0,0.0,0.0003726810866737651,320,80,87.12945866584778,1.0891182333230973,0.27227955833077433,0.0932244597002864,0.002895271330635296,0.6430459570903622,1.4183609434681444e-05,0.0074269796255975965,0.0,0.0,0.0,0.0,0.002975074338610284,80,20,19.336661100387573,0.9668330550193787,0.24170826375484467,0.051954638119786976 | |
| 27,0.00045546451366362817,0.07809749107524873,1.9649450981934686e-07,0.009275437286123633,0.0,0.0,0.0,0.0,0.00046089367940567173,320,80,86.8785035610199,1.0859812945127487,0.2714953236281872,0.09191167326644063,0.0028493454796262086,0.6770066560213991,1.2013152249146231e-05,0.007457524072378874,0.0,0.0,0.0,0.0,0.0029267561651067807,80,20,19.251688241958618,0.9625844120979309,0.24064610302448272,0.04829146796837449 | |
| 28,0.00033174752630884543,0.06612742295388116,1.180179221131573e-07,0.009398166846949607,0.0,0.0,0.0,0.0,0.0003358864170422748,320,80,86.8982789516449,1.0862284868955612,0.2715571217238903,0.09198763176100329,0.0027787499493570067,0.7222041939938209,1.1131508226305125e-05,0.0075377147179096935,0.0,0.0,0.0,0.0,0.0028551170726132113,80,20,19.059608459472656,0.9529804229736328,0.2382451057434082,0.04740339070558548 | |
| 29,0.0003597953202529425,0.024496306587910778,1.142882032886816e-07,0.009390755963977426,0.0,0.0,0.0,0.0,0.00036418304370045007,320,80,86.48576998710632,1.081072124838829,0.27026803120970727,0.09743952928110958,0.0026345706894062458,0.7635843992035003,1.1359682959266593e-05,0.007559091807343066,0.0,0.0,0.0,0.0,0.0027058403313276356,80,20,18.992973566055298,0.9496486783027649,0.23741216957569122,0.046653942298144103 | |
| 30,0.0004039141500470578,0.09576162853820311,1.8995679496383588e-07,0.009361328079830856,0.0,0.0,0.0,0.0,0.00040872053962175413,320,80,86.09150242805481,1.076143780350685,0.26903594508767126,0.09370959255611525,0.0027511596577824093,0.6731958146176111,1.417353958319545e-05,0.007637712243013084,0.0,0.0,0.0,0.0,0.0028271504568692764,80,20,18.946882724761963,0.9473441362380981,0.23683603405952453,0.047970831673592326 | |
| 31,0.00030792704831128503,0.03007892958051279,1.6044353357220544e-07,0.009489066211972385,0.0,0.0,0.0,0.0,0.00031191278568485357,320,80,86.08254933357239,1.076031866669655,0.2690079666674137,0.09290082987863571,0.0027197946154046805,0.6787573918956695,9.246046926492114e-06,0.007403567247092724,0.0,0.0,0.0,0.0,0.002790288768301252,80,20,19.061045169830322,0.9530522584915161,0.23826306462287902,0.04738053930923343 | |
| 32,0.0002918601881447103,0.02865394608418743,1.377780366400853e-07,0.009482038370333613,0.0,0.0,0.0,0.0,0.00029560920602307307,320,80,85.58978748321533,1.0698723435401916,0.2674680858850479,0.09774628963787109,0.002647562528727576,0.6132856009367742,1.2823440413711751e-05,0.007520708185620606,0.0,0.0,0.0,0.0,0.0027170190507604273,80,20,18.80093550682068,0.940046775341034,0.2350116938352585,0.0500286478549242 | |