,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.020646917774962883,0.7029616862738138,0.0014198488189189598,0.0,0.0,0.0,0.0,0.0,0.020646917774962883,900,225,211.21505284309387,0.9387335681915283,0.23468339204788208,0.12668553197549448,0.00800067097414285,0.8323721770502058,0.00025833536566152146,0.0,0.0,0.0,0.0,0.0,0.00800067097414285,450,113,91.81465721130371,0.8125190903655196,0.20403257158067492,0.07484065717399384
1,0.007335189075060447,0.5605360431658958,0.00016509135985034204,0.0,0.0,0.0,0.0,0.0,0.007335189075060447,900,225,211.18131518363953,0.9385836230383979,0.23464590575959948,0.09665914196934965,0.005770373049502572,1.328719814272311,0.0001455103024895009,0.0,0.0,0.0,0.0,0.0,0.005770373049502572,450,113,92.31423711776733,0.8169401514846667,0.20514274915059408,0.06159803802889269
2,0.004101434572932905,0.40740934907574267,4.4437185304309564e-05,0.0,0.0,0.0,0.0,0.0,0.004101434572932905,900,225,211.43363165855408,0.9397050295935737,0.23492625739839343,0.10036693361898263,0.005793615489771279,2.93090469723272,7.996645985781352e-05,0.0,0.0,0.0,0.0,0.0,0.005793615489771279,450,113,92.21822166442871,0.8160904572073338,0.20492938147650824,0.0686845871407654
3,0.0032525908350200753,0.45932424604433697,1.5064653623929553e-05,0.0,0.0,0.0,0.0,0.0,0.0032525908350200753,900,225,210.6352882385254,0.9361568366156684,0.2340392091539171,0.09941185830367937,0.0025840073977209006,2.9610207560058215,8.98458365437745e-06,0.0,0.0,0.0,0.0,0.0,0.0025840073977209006,450,113,91.6226761341095,0.81082014277973,0.20360594696468778,0.04679510168797147
4,0.0028994390259807308,0.27792873476118807,1.2238399814356409e-05,0.0,0.0,0.0,0.0,0.0,0.0028994390259807308,900,225,210.90299940109253,0.9373466640048557,0.23433666600121392,0.10629830273903078,0.004226378290137897,3.139442252464839,2.8896792241464515e-05,0.0,0.0,0.0,0.0,0.0,0.004226378290137897,450,113,92.30420923233032,0.8168514091356666,0.20512046496073405,0.04041025609363167
5,0.0029357373788823477,0.35013624048834924,1.5252040759503315e-05,0.0,0.0,0.0,0.0,0.0,0.0029357373788823477,900,225,210.8098328113556,0.9369325902726915,0.23423314756817287,0.10065761231092943,0.003366937771077371,2.7330855187934775,2.1526505657937356e-05,0.0,0.0,0.0,0.0,0.0,0.003366937771077371,450,113,92.45483732223511,0.8181844010817266,0.20545519404941134,0.048643345435137604
6,0.0025804581021010463,0.33193543293685124,1.1130432750345e-05,0.0,0.0,0.0,0.0,0.0,0.0025804581021010463,900,225,210.82577991485596,0.9370034662882487,0.23425086657206218,0.10346181529677577,0.0029045950072920986,4.791847205054414,1.7926797626652248e-05,0.0,0.0,0.0,0.0,0.0,0.0029045950072920986,450,113,92.71966814994812,0.8205280367252046,0.20604370699988472,0.04583418705378066
7,0.0024444010488999385,0.27924930442016427,1.0806890312169106e-05,0.0,0.0,0.0,0.0,0.0,0.0024444010488999385,900,225,211.59207558631897,0.9404092248280843,0.23510230620702108,0.10225464255238573,0.002720622533104486,3.2791891266000865,1.838483538128186e-05,0.0,0.0,0.0,0.0,0.0,0.002720622533104486,450,113,96.57148313522339,0.8546148950019768,0.21460329585605198,0.04418732333965435
8,0.00242780985414154,0.25769682647126535,1.037415603605397e-05,0.0,0.0,0.0,0.0,0.0,0.00242780985414154,900,225,213.52410340309143,0.9489960151248508,0.2372490037812127,0.10494643683855732,0.002887863010659607,1.8791022815315657,2.555802672021092e-05,0.0,0.0,0.0,0.0,0.0,0.002887863010659607,450,113,92.99426054954529,0.8229580579605777,0.20665391233232286,0.059518284711418096
9,0.002208985082106665,0.42546326303130916,7.684016633753033e-06,0.0,0.0,0.0,0.0,0.0,0.002208985082106665,900,225,213.6311333179474,0.9494717036353217,0.23736792590883043,0.10293637524772849,0.0024638745025731624,3.782287786237916,1.655159604749657e-05,0.0,0.0,0.0,0.0,0.0,0.0024638745025731624,450,113,97.28867506980896,0.8609617262814953,0.21619705571068656,0.04808065608046965
10,0.0020764596655175813,0.3896139937922279,7.334516867581215e-06,0.0,0.0,0.0,0.0,0.0,0.0020764596655175813,900,225,221.03631401062012,0.9823836178249783,0.24559590445624457,0.10016421435814765,0.0025203956082178692,2.2354346593865277,2.0540576969837897e-05,0.0,0.0,0.0,0.0,0.0,0.0025203956082178692,450,113,97.72019529342651,0.8647804893223585,0.2171559895409478,0.057214381701758014