,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.03644084440955699,6.612403465312778,0.0045574880142667225,0.0,0.0,0.0,0.0,0.0,0.03644084440955699,320,160,166.3736925125122,1.0398355782032014,0.5199177891016007,0.07979056920851804,0.007586553805595031,3.274879661123663,0.00010986064107032512,0.0,0.0,0.0,0.0,0.0,0.007586553805595031,80,40,34.049020767211914,0.8512255191802979,0.42561275959014894,0.04524085290104267
1,0.010942080118709896,3.6911100070230938,0.00037560693868070696,0.0,0.0,0.0,0.0,0.0,0.010942080118709896,320,160,166.92550325393677,1.0432843953371047,0.5216421976685524,0.14872968647239873,0.026652724126324755,7.903173122670358,0.0026581332727844825,0.0,0.0,0.0,0.0,0.0,0.026652724126324755,80,40,33.64800572395325,0.8412001430988312,0.4206000715494156,0.02281103633413295
2,0.010455712701009312,3.9791094253597508,0.0003505937737199728,0.0,0.0,0.0,0.0,0.0,0.010455712701009312,320,160,166.77985906600952,1.0423741191625595,0.5211870595812798,0.15552815834055309,0.012862359535210999,5.169579194596042,0.000530584767799982,0.0,0.0,0.0,0.0,0.0,0.012862359535210999,80,40,33.91816258430481,0.8479540646076202,0.4239770323038101,0.026617402605188543
3,0.007001378159810656,2.796910241492641,0.00011505681204335635,0.0,0.0,0.0,0.0,0.0,0.007001378159810656,320,160,166.83584594726562,1.04272403717041,0.521362018585205,0.13855754688775052,0.004590427323637414,6.018633032932655,3.031255504705524e-05,0.0,0.0,0.0,0.0,0.0,0.004590427323637414,80,40,33.972458839416504,0.8493114709854126,0.4246557354927063,0.04482766297978742
4,0.0072793409375947245,2.296859505424894,0.0001573822781525827,0.0,0.0,0.0,0.0,0.0,0.0072793409375947245,320,160,167.01165390014648,1.0438228368759155,0.5219114184379577,0.16224303948229135,0.004423623792263243,6.739300958566085,6.933249757632432e-05,0.0,0.0,0.0,0.0,0.0,0.004423623792263243,80,40,33.651365756988525,0.8412841439247132,0.4206420719623566,0.026270963538991055
5,0.006381618711196779,2.8839405857504543,0.00012163765120135503,0.0,0.0,0.0,0.0,0.0,0.006381618711196779,320,160,170.09552717208862,1.063097044825554,0.531548522412777,0.14938008590495427,0.0050863039046817,7.422684189675602,4.899889018377124e-05,0.0,0.0,0.0,0.0,0.0,0.0050863039046817,80,40,37.50921392440796,0.937730348110199,0.4688651740550995,0.029683142538488028
6,0.006045459082537263,2.934065179698594,8.634548304524475e-05,0.0,0.0,0.0,0.0,0.0,0.006045459082537263,320,160,173.5575668811798,1.0847347930073739,0.5423673965036869,0.128542572739525,0.004006205599580426,6.11909287295653,2.419293156137453e-05,0.0,0.0,0.0,0.0,0.0,0.004006205599580426,80,40,36.28923416137695,0.9072308540344238,0.4536154270172119,0.02928218668603222
7,0.005807553204306259,3.233931762049849,6.728753389705139e-05,0.0,0.0,0.0,0.0,0.0,0.005807553204306259,320,160,173.0330581665039,1.0814566135406494,0.5407283067703247,0.13020459838949136,0.004725078083447442,5.915451907179363,4.036447823969336e-05,0.0,0.0,0.0,0.0,0.0,0.004725078083447442,80,40,35.27379822731018,0.8818449556827546,0.4409224778413773,0.03375284938047116
8,0.005597162095671138,2.4687882317225887,9.451023661321612e-05,0.0,0.0,0.0,0.0,0.0,0.005597162095671138,320,160,171.33222126960754,1.070826382935047,0.5354131914675235,0.16350435888944048,0.005341960320765793,4.580611580479113,5.5816291564272924e-05,0.0,0.0,0.0,0.0,0.0,0.005341960320765793,80,40,35.73819422721863,0.8934548556804657,0.44672742784023284,0.028984847072570118
9,0.005475803721810735,2.9333888558279755,5.944834627596224e-05,0.0,0.0,0.0,0.0,0.0,0.005475803721810735,320,160,181.32838702201843,1.1333024188876153,0.5666512094438076,0.14286351032374114,0.00505017776886234,5.1855854878382335,4.6662756986093344e-05,0.0,0.0,0.0,0.0,0.0,0.00505017776886234,80,40,39.708540201187134,0.9927135050296784,0.4963567525148392,0.028717920677536313
10,0.005619597199086002,2.6044400273167847,0.0001110976432221262,0.0,0.0,0.0,0.0,0.0,0.005619597199086002,320,160,191.31377625465393,1.1957111015915871,0.5978555507957936,0.1449692299744129,0.003595894821683032,4.273874850746305,3.0524735783286906e-05,0.0,0.0,0.0,0.0,0.0,0.003595894821683032,80,40,40.169761657714844,1.004244041442871,0.5021220207214355,0.038896191439198445
11,0.005651603202302624,2.1689565299521996,8.184791047085491e-05,0.0,0.0,0.0,0.0,0.0,0.005651603202302624,320,160,191.00746726989746,1.1937966704368592,0.5968983352184296,0.13332524879906488,0.003731331395374582,4.425117476265046,2.9901788849429067e-05,0.0,0.0,0.0,0.0,0.0,0.003731331395374582,80,40,40.135857343673706,1.0033964335918426,0.5016982167959213,0.04076760089155869
12,0.005675629577149266,1.5336641537275533,8.70407526315881e-05,0.0,0.0,0.0,0.0,0.0,0.005675629577149266,320,160,191.66054034233093,1.1978783771395682,0.5989391885697841,0.16240290804776122,0.0038316375943395543,4.709503752670197,2.2613220733549987e-05,0.0,0.0,0.0,0.0,0.0,0.0038316375943395543,80,40,40.268686056137085,1.0067171514034272,0.5033585757017136,0.028456049150554462
13,0.00614657213177452,2.04597273792412,7.693676020577578e-05,0.0,0.0,0.0,0.0,0.0,0.00614657213177452,320,160,189.26595377922058,1.1829122111201287,0.5914561055600643,0.15553884600512902,0.0042953793235938065,4.08081223766667,3.2513099143405276e-05,0.0,0.0,0.0,0.0,0.0,0.0042953793235938065,80,40,40.096407890319824,1.0024101972579955,0.5012050986289978,0.02533484929444967
14,0.005231396525572052,1.73261989282967,7.520394053998295e-05,0.0,0.0,0.0,0.0,0.0,0.005231396525572052,320,160,191.00612378120422,1.1937882736325265,0.5968941368162632,0.1496530410122432,0.003655807935683697,4.01203602381832,2.1951282996873765e-05,0.0,0.0,0.0,0.0,0.0,0.003655807935683697,80,40,40.18678307533264,1.004669576883316,0.502334788441658,0.03820230678429652
15,0.005476373057177852,2.6803855865461252,8.069490082747689e-05,0.0,0.0,0.0,0.0,0.0,0.005476373057177852,320,160,191.20353507995605,1.1950220942497254,0.5975110471248627,0.12890588956015564,0.0038699784756772715,5.125400327792415,2.3996304280442248e-05,0.0,0.0,0.0,0.0,0.0,0.0038699784756772715,80,40,39.97018790245056,0.9992546975612641,0.49962734878063203,0.03100545472552767