,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.22352998348069378,80.13195664776967,0.09520609854183135,0.0,0.0,0.0,0.0,0.0,0.22352998348069378,320,80,103.12237620353699,1.2890297025442123,0.3222574256360531,0.0988283173581749,0.01245800144970417,1.6089594815347597,0.00019736949793127678,0.0,0.0,0.0,0.0,0.0,0.01245800144970417,80,20,20.080093383789062,1.004004669189453,0.25100116729736327,0.07783476307522505
1,0.008753520530444803,0.34713386677235575,0.00011480308697381734,0.0,0.0,0.0,0.0,0.0,0.008753520530444803,320,80,103.45079112052917,1.2931348890066148,0.3232837222516537,0.19368809863808564,0.007529928936855867,3.185711348353652,0.00010870498384889516,0.0,0.0,0.0,0.0,0.0,0.007529928936855867,80,20,20.23315191268921,1.0116575956344604,0.2529143989086151,0.043344746553339066
2,0.006802400949891307,0.3522556849198281,0.0001498469549909341,0.0,0.0,0.0,0.0,0.0,0.006802400949891307,320,80,103.24245595932007,1.290530699491501,0.3226326748728752,0.18977850895607845,0.007473361069423845,4.433246247235365,0.00010783077031044641,0.0,0.0,0.0,0.0,0.0,0.007473361069423845,80,20,20.317150354385376,1.0158575177192688,0.2539643794298172,0.043154743919149044
3,0.008915021323991823,0.718201418556464,8.08643664615523e-05,0.0,0.0,0.0,0.0,0.0,0.008915021323991823,320,80,103.76086711883545,1.2970108389854431,0.3242527097463608,0.18378724994836376,0.015541119105182587,4.1963203363062345,0.0004786531295351892,0.0,0.0,0.0,0.0,0.0,0.015541119105182587,80,20,20.26967740058899,1.0134838700294495,0.2533709675073624,0.03725434660445899
4,0.006931817024451448,0.35298403866354533,7.142922729861737e-05,0.0,0.0,0.0,0.0,0.0,0.006931817024451448,320,80,103.68316864967346,1.2960396081209182,0.32400990203022956,0.1754610677191522,0.006783264396653976,2.084030251805075,0.00010564910719947917,0.0,0.0,0.0,0.0,0.0,0.006783264396653976,80,20,20.532756090164185,1.0266378045082092,0.2566594511270523,0.046784830396063626
5,0.005099834372958867,0.0875181610394841,9.274947589943961e-05,0.0,0.0,0.0,0.0,0.0,0.005099834372958867,320,80,103.51723384857178,1.2939654231071471,0.3234913557767868,0.19479809664189815,0.006807428642059676,6.448283100366529,8.50121301514406e-05,0.0,0.0,0.0,0.0,0.0,0.006807428642059676,80,20,20.114439249038696,1.005721962451935,0.2514304906129837,0.04373221881105564
6,0.00480109396030457,0.9528829011607514,5.0811379982423e-05,0.0,0.0,0.0,0.0,0.0,0.00480109396030457,320,80,103.59416389465332,1.2949270486831665,0.32373176217079164,0.178886199297267,0.006987621737061999,4.279015872643868,9.801611965674085e-05,0.0,0.0,0.0,0.0,0.0,0.006987621737061999,80,20,20.27844500541687,1.0139222502708436,0.2534805625677109,0.04849170843372121
7,0.00464072308905088,0.1356045210827208,6.834771834407436e-05,0.0,0.0,0.0,0.0,0.0,0.00464072308905088,320,80,103.63158965110779,1.2953948706388474,0.32384871765971185,0.19017753867083229,0.006893738606595434,3.8557679186087626,8.831684561805275e-05,0.0,0.0,0.0,0.0,0.0,0.006893738606595434,80,20,20.594661712646484,1.0297330856323241,0.25743327140808103,0.04350378216477111
8,0.004308880444841634,0.06846157661166216,5.0194533013741347e-05,0.0,0.0,0.0,0.0,0.0,0.004308880444841634,320,80,103.90516233444214,1.2988145291805266,0.32470363229513166,0.18733386998064816,0.006929469533497467,3.2670128452096834,7.968755999527843e-05,0.0,0.0,0.0,0.0,0.0,0.006929469533497467,80,20,20.433101177215576,1.021655058860779,0.2554137647151947,0.04181941950228065
9,0.004508837422326906,0.08966287379656705,8.609927907704219e-05,0.0,0.0,0.0,0.0,0.0,0.004508837422326906,320,80,103.64076137542725,1.2955095171928406,0.32387737929821014,0.18831826079403982,0.0070835084756254215,4.297422426286471,0.00011362928610676448,0.0,0.0,0.0,0.0,0.0,0.0070835084756254215,80,20,20.304687023162842,1.015234351158142,0.2538085877895355,0.0519675396499224
10,0.004514601970731747,0.1381884859496653,8.4598984369378e-05,0.0,0.0,0.0,0.0,0.0,0.004514601970731747,320,80,103.90474796295166,1.2988093495368958,0.32470233738422394,0.17621430779545336,0.005505984234332573,5.328735202133521,3.370500902892815e-05,0.0,0.0,0.0,0.0,0.0,0.005505984234332573,80,20,20.54746174812317,1.0273730874061584,0.2568432718515396,0.051376725709997115
11,0.004419548849091371,0.07128540304256603,6.623427232033962e-05,0.0,0.0,0.0,0.0,0.0,0.004419548849091371,320,80,103.64773535728455,1.2955966919660569,0.3238991729915142,0.18890725779347123,0.005045573477400467,4.834901636225572,2.2054046640818116e-05,0.0,0.0,0.0,0.0,0.0,0.005045573477400467,80,20,20.591118812561035,1.0295559406280517,0.2573889851570129,0.051232723612338306
12,0.004236863098185495,0.04387387494761443,8.839865267179564e-05,0.0,0.0,0.0,0.0,0.0,0.004236863098185495,320,80,103.94386696815491,1.2992983371019364,0.3248245842754841,0.1935716205276549,0.008129652022034861,5.03432033594964,0.00015811533519221043,0.0,0.0,0.0,0.0,0.0,0.008129652022034861,80,20,20.881214380264282,1.044060719013214,0.2610151797533035,0.04673133364703972
13,0.004276435652536747,0.0654336524629164,4.581930034836763e-05,0.0,0.0,0.0,0.0,0.0,0.004276435652536747,320,80,103.00667309761047,1.287583413720131,0.32189585343003274,0.18084844152908772,0.007877110847039149,2.788165700972968,0.0001813338503336759,0.0,0.0,0.0,0.0,0.0,0.007877110847039149,80,20,20.325989723205566,1.0162994861602783,0.2540748715400696,0.04980120111722499
14,0.0038373295942619734,0.0791764844770995,4.599695211216274e-05,0.0,0.0,0.0,0.0,0.0,0.0038373295942619734,320,80,103.24337792396545,1.290542224049568,0.322635556012392,0.18313483651727439,0.0050060375331668185,3.8634611237153877,2.9480706338880223e-05,0.0,0.0,0.0,0.0,0.0,0.0050060375331668185,80,20,20.198811054229736,1.0099405527114869,0.2524851381778717,0.058232192660216245
15,0.003900589924239739,0.10312648875277333,2.1983545730452913e-05,0.0,0.0,0.0,0.0,0.0,0.003900589924239739,320,80,103.35287404060364,1.2919109255075454,0.32297773137688635,0.19720614301040768,0.009117326361592858,1.760603382944828,0.00027292398581290066,0.0,0.0,0.0,0.0,0.0,0.009117326361592858,80,20,20.001904249191284,1.0000952124595641,0.25002380311489103,0.048848784435540436
16,0.0018584069126518442,0.04251637269783757,1.0266101569923053e-05,0.0,0.0,0.0,0.0,0.0,0.0018584069126518442,320,80,102.94779467582703,1.286847433447838,0.3217118583619595,0.18552915730979294,0.007703973473689984,1.3023191384279245,0.00017489787961899594,0.0,0.0,0.0,0.0,0.0,0.007703973473689984,80,20,20.37928342819214,1.018964171409607,0.25474104285240173,0.05299922423437238
17,0.0008170823710770492,0.018880133842297652,5.023515140430146e-07,0.0,0.0,0.0,0.0,0.0,0.0008170823710770492,320,80,102.69966387748718,1.2837457984685898,0.32093644961714746,0.1925133554963395,0.00931795308351866,2.071352872970965,0.0002587945756321958,0.0,0.0,0.0,0.0,0.0,0.00931795308351866,80,20,19.96341824531555,0.9981709122657776,0.2495427280664444,0.053129641944542526
18,0.0003576292982074847,0.012536979420885785,4.498594921749366e-08,0.0,0.0,0.0,0.0,0.0,0.0003576292982074847,320,80,100.82452273368835,1.2603065341711044,0.3150766335427761,0.19631691183894873,0.00876514861229225,1.4645986258908124,0.0002133372895583463,0.0,0.0,0.0,0.0,0.0,0.00876514861229225,80,20,19.451266765594482,0.9725633382797241,0.24314083456993102,0.05004617176018655
19,0.0002510753680212474,0.008272775144363465,3.3253448800756014e-08,0.0,0.0,0.0,0.0,0.0,0.0002510753680212474,320,80,102.30520558357239,1.2788150697946548,0.3197037674486637,0.19034422542899848,0.00618170693560387,1.5110018466951716,7.781338554512241e-05,0.0,0.0,0.0,0.0,0.0,0.00618170693560387,80,20,19.740620136260986,0.9870310068130493,0.24675775170326233,0.055138330021873114
20,0.00028350398017664704,0.019687367545015277,2.482109546082703e-07,0.0,0.0,0.0,0.0,0.0,0.00028350398017664704,320,80,99.86743569374084,1.2483429461717606,0.31208573654294014,0.1864254915737547,0.007587061779486248,1.3006179411045196,0.00014909935981792798,0.0,0.0,0.0,0.0,0.0,0.007587061779486248,80,20,20.21121335029602,1.0105606675148011,0.2526401668787003,0.05435547353699803
21,0.00025932476952164054,0.007903821782640907,9.45318477345975e-10,0.0,0.0,0.0,0.0,0.0,0.00025932476952164054,320,80,103.72837948799133,1.2966047435998918,0.32415118589997294,0.18518321572337298,0.006430168935912662,1.9212478918598208,9.073310130256474e-05,0.0,0.0,0.0,0.0,0.0,0.006430168935912662,80,20,19.81023097038269,0.9905115485191345,0.24762788712978362,0.05623150994069874
22,0.00033484522286357786,0.012371280277519502,1.6720436467560026e-08,0.0,0.0,0.0,0.0,0.0,0.00033484522286357786,320,80,102.39797282218933,1.2799746602773667,0.3199936650693417,0.1927722441148944,0.0066237156010174655,1.626585018528567,9.447487505163111e-05,0.0,0.0,0.0,0.0,0.0,0.0066237156010174655,80,20,20.115633487701416,1.0057816743850707,0.2514454185962677,0.054200840881094337
23,0.00015932412852635026,0.025226200853674642,3.639718875007858e-08,0.0,0.0,0.0,0.0,0.0,0.00015932412852635026,320,80,103.57392120361328,1.294674015045166,0.3236685037612915,0.1952298643416725,0.0065823450138850605,1.5178716897570212,9.93379512048212e-05,0.0,0.0,0.0,0.0,0.0,0.0065823450138850605,80,20,21.15859341621399,1.0579296708106996,0.2644824177026749,0.0543066727463156
24,0.00012041164002258853,0.006688666018078895,1.577604157902094e-08,0.0,0.0,0.0,0.0,0.0,0.00012041164002258853,320,80,107.04027843475342,1.3380034804344176,0.3345008701086044,0.18686838666908442,0.007150729962449987,1.6130354540884582,0.00011762036998774761,0.0,0.0,0.0,0.0,0.0,0.007150729962449987,80,20,21.25710916519165,1.0628554582595826,0.26571386456489565,0.050873439060524106
25,8.769563716697349e-05,0.005887569199180521,3.494414894220867e-09,0.0,0.0,0.0,0.0,0.0,8.769563716697349e-05,320,80,106.57941937446594,1.3322427421808243,0.33306068554520607,0.18599217470618895,0.00832268671510974,1.4815574481464182,0.00019259734704257792,0.0,0.0,0.0,0.0,0.0,0.00832268671510974,80,20,20.612091302871704,1.0306045651435851,0.2576511412858963,0.05257055321708322
26,8.882110219019524e-05,0.002085926350794054,2.3958830939162234e-09,0.0,0.0,0.0,0.0,0.0,8.882110219019524e-05,320,80,106.64562821388245,1.3330703526735306,0.33326758816838264,0.20409600271377712,0.006672654673457146,1.4586342717834213,9.671619951117094e-05,0.0,0.0,0.0,0.0,0.0,0.006672654673457146,80,20,20.21643877029419,1.0108219385147095,0.25270548462867737,0.0529700854793191
27,7.043356762892472e-05,0.005430679229713497,4.15760022837944e-10,0.0,0.0,0.0,0.0,0.0,7.043356762892472e-05,320,80,102.27982902526855,1.278497862815857,0.31962446570396424,0.2028546938439831,0.006019308035320137,1.3263217964900833,7.28448786667002e-05,0.0,0.0,0.0,0.0,0.0,0.006019308035320137,80,20,20.296292066574097,1.0148146033287049,0.2537036508321762,0.05590685121715069