polarolouis/REE-RL-Lola
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Some results

Browse source
This commit is contained in:
Louis Lacoste 2025-12-07 17:39:32 +01:00
parent 1506bdaab1
commit 905ed01153
5 changed files with 16 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
results/best_models.csv
View file

@ -1,3 +1,3 @@
participant_id,model,BIC,converged
167y1h5i,GAIN_LOSS,1051.8545061869925,TRUE
1e1vfxfy,GAIN_LOSS,929.6450964059819,TRUE
participant,model,BIC,AIC,negLL
l5hj4bqu,REE_BIASED_SIMPLE,330.06777017606026,306.11898289341235,147.05949144670618
qfmtmjjy,REE_BIASED_SIMPLE,397.8509544006175,373.9021671179696,180.9510835589848

1 participant_id participant model BIC converged AIC negLL
2 167y1h5i l5hj4bqu GAIN_LOSS REE_BIASED_SIMPLE 1051.8545061869925 330.06777017606026 TRUE 306.11898289341235 147.05949144670618
3 1e1vfxfy qfmtmjjy GAIN_LOSS REE_BIASED_SIMPLE 929.6450964059819 397.8509544006175 TRUE 373.9021671179696 180.9510835589848

7
results/global_comparison.csv
View file

@ -1,3 +1,4 @@
model,n_params,n_converged,mean_negLL,total_negLL,total_AIC,total_BIC,mean_convergence_range
GAIN_LOSS,4,2,483.39197155402763,966.7839431080553,1949.5678862161105,1981.4996025929745,4.615930038198712e-8
HOMOGENEOUS,3,2,539.752516085906,1079.505032171812,2171.010064343624,2194.958851626272,2.0892058557819837e-5
model,n_params,n_converged,n_participants,mean_negLL,total_negLL,total_AIC,total_BIC
REE_BIASED_SIMPLE,6,2,2,164.00528750284548,328.01057500569095,680.0211500113819,727.9187245766777
GAIN_LOSS,4,2,2,265.25670321172487,530.5134064234497,1077.0268128468995,1108.9585292237634
HOMOGENEOUS,3,2,2,351.61199392781214,703.2239878556243,1418.4479757112485,1442.3967629938966

1 model n_params n_converged n_participants mean_negLL total_negLL total_AIC mean_convergence_range total_BIC
2 GAIN_LOSS REE_BIASED_SIMPLE 4 6 2 2 483.39197155402763 164.00528750284548 966.7839431080553 328.01057500569095 1949.5678862161105 680.0211500113819 4.615930038198712e-8 1981.4996025929745 727.9187245766777
3 HOMOGENEOUS GAIN_LOSS 3 4 2 2 539.752516085906 265.25670321172487 1079.505032171812 530.5134064234497 2171.010064343624 1077.0268128468995 2.0892058557819837e-5 2194.958851626272 1108.9585292237634
4 HOMOGENEOUS 3 2 2 351.61199392781214 703.2239878556243 1418.4479757112485 1442.3967629938966

6
results/results_GAIN_LOSS.csv
View file

@ -1,3 +1,3 @@
participant_id,model,n_params,negLL,AIC,BIC,convergence_sd,convergence_range,hessian_positive_definite,converged,has_BS_seen,has_JP_seen,alpha_loss,se_alpha_loss,alpha_gain,se_alpha_gain,forget,se_forget,lambda,se_lambda
167y1h5i,GAIN_LOSS,4,513.9443239992803,1035.8886479985606,1051.8545061869925,1.6892161240846193e-9,4.670710040954873e-9,TRUE,TRUE,FALSE,FALSE,-4.999999999572799,0.43882834813478433,4.999999999974559,0.33323708070269153,4.999999999930331,0.30695089661324654,-1.4051507158699554,0.09652272582280916
1e1vfxfy,GAIN_LOSS,4,452.83961910877497,913.6792382175499,929.6450964059819,3.735146110929026e-8,8.764789072301937e-8,TRUE,TRUE,FALSE,FALSE,-4.999999999984108,0.45214712818390496,4.999999999747621,0.4486530859417248,4.999999999941144,0.3096072666178563,-1.3143510245415637,0.06931889910013954
model,n_params,negLL,AIC,BIC,ELBO,ELBO_SD,n_iterations,converged,method,alpha_loss,sd_alpha_loss,alpha_gain,sd_alpha_gain,forget,sd_forget,lambda,sd_lambda,participant
GAIN_LOSS,4,288.3257712089803,584.6515424179606,600.6174006063926,-293.8098204601751,0.018419418023481016,39,True,VBMC,-4.789747942697576,0.1852296017413521,4.910180863178969,0.0920756887226078,4.91788510293336,0.07532131905941286,-0.6659006141798032,0.06651084767459509,qfmtmjjy
GAIN_LOSS,4,242.18763521446942,492.37527042893885,508.34112861737077,-243.60122835783287,0.027016475818384324,55,True,VBMC,-4.821632021951461,0.14164524805583711,4.268359226923721,0.4216671818661271,4.234807151138014,0.7157080121250583,-0.0024350764545131897,0.06697639216935664,l5hj4bqu

1 participant_id model n_params negLL AIC BIC ELBO convergence_sd ELBO_SD convergence_range n_iterations hessian_positive_definite converged method has_BS_seen alpha_loss has_JP_seen sd_alpha_loss alpha_gain se_alpha_loss sd_alpha_gain forget se_alpha_gain sd_forget lambda se_forget sd_lambda participant se_lambda
2 167y1h5i GAIN_LOSS 4 513.9443239992803 288.3257712089803 1035.8886479985606 584.6515424179606 1051.8545061869925 600.6174006063926 -293.8098204601751 1.6892161240846193e-9 0.018419418023481016 4.670710040954873e-9 39 TRUE TRUE True VBMC FALSE -4.999999999572799 -4.789747942697576 FALSE 0.1852296017413521 4.999999999974559 4.910180863178969 0.43882834813478433 0.0920756887226078 4.999999999930331 4.91788510293336 0.33323708070269153 0.07532131905941286 -1.4051507158699554 -0.6659006141798032 0.30695089661324654 0.06651084767459509 qfmtmjjy 0.09652272582280916
3 1e1vfxfy GAIN_LOSS 4 452.83961910877497 242.18763521446942 913.6792382175499 492.37527042893885 929.6450964059819 508.34112861737077 -243.60122835783287 3.735146110929026e-8 0.027016475818384324 8.764789072301937e-8 55 TRUE TRUE True VBMC FALSE -4.999999999984108 -4.821632021951461 FALSE 0.14164524805583711 4.999999999747621 4.268359226923721 0.45214712818390496 0.4216671818661271 4.999999999941144 4.234807151138014 0.4486530859417248 0.7157080121250583 -1.3143510245415637 -0.0024350764545131897 0.3096072666178563 0.06697639216935664 l5hj4bqu 0.06931889910013954

6
results/results_HOMOGENEOUS.csv
View file

@ -1,3 +1,3 @@
participant_id,model,n_params,negLL,AIC,BIC,convergence_sd,convergence_range,hessian_positive_definite,converged,has_BS_seen,has_JP_seen,alpha,se_alpha,forget,se_forget,lambda,se_lambda
167y1h5i,HOMOGENEOUS,3,537.005463940143,1080.010927880286,1091.98532152161,2.9700346380225555e-12,6.707523425575346e-12,FALSE,TRUE,FALSE,FALSE,-4.999999999999991,NA,-4.291766151436294,NA,-2.999999999999999,NA
1e1vfxfy,HOMOGENEOUS,3,542.499568231669,1090.999136463338,1102.973530104662,1.8563286694240588e-5,4.178411040811625e-5,TRUE,TRUE,FALSE,FALSE,-0.7330899515496483,0.2661977203672045,4.999999999999953,0.34607559474050387,-0.7373357490749459,0.14111490702769247
model,n_params,negLL,AIC,BIC,ELBO,ELBO_SD,n_iterations,converged,method,alpha,sd_alpha,forget,sd_forget,lambda,sd_lambda,participant
HOMOGENEOUS,3,473.54186038059424,953.0837207611885,965.0581144025124,-474.52031064945436,0.0027499883551695237,23,True,VBMC,-4.847809870984468,0.1300398091269694,3.3569794305963523,1.038293691574052,-1.606519382201114,0.15587842127930715,qfmtmjjy
HOMOGENEOUS,3,229.68212747503006,465.3642549500601,477.33864859138407,-233.27227846817146,0.002332084917703993,26,True,VBMC,-4.937369596752305,0.057725958053491,-4.194101183136464,0.40801806005370317,-1.8334356471237399,0.09190552874296334,l5hj4bqu

1 participant_id model n_params negLL AIC BIC ELBO convergence_sd ELBO_SD convergence_range n_iterations hessian_positive_definite converged method has_BS_seen alpha has_JP_seen sd_alpha forget se_alpha sd_forget lambda se_forget sd_lambda participant se_lambda
2 167y1h5i HOMOGENEOUS 3 537.005463940143 473.54186038059424 1080.010927880286 953.0837207611885 1091.98532152161 965.0581144025124 -474.52031064945436 2.9700346380225555e-12 0.0027499883551695237 6.707523425575346e-12 23 FALSE TRUE True VBMC FALSE -4.999999999999991 -4.847809870984468 FALSE 0.1300398091269694 -4.291766151436294 3.3569794305963523 NA 1.038293691574052 -2.999999999999999 -1.606519382201114 NA 0.15587842127930715 qfmtmjjy NA
3 1e1vfxfy HOMOGENEOUS 3 542.499568231669 229.68212747503006 1090.999136463338 465.3642549500601 1102.973530104662 477.33864859138407 -233.27227846817146 1.8563286694240588e-5 0.002332084917703993 4.178411040811625e-5 26 TRUE TRUE True VBMC FALSE -0.7330899515496483 -4.937369596752305 FALSE 0.057725958053491 4.999999999999953 -4.194101183136464 0.2661977203672045 0.40801806005370317 -0.7373357490749459 -1.8334356471237399 0.34607559474050387 0.09190552874296334 l5hj4bqu 0.14111490702769247

3
results/results_REE_BIASED_SIMPLE.csv Normal file
View file

@ -0,0 +1,3 @@
model,n_params,negLL,AIC,BIC,ELBO,ELBO_SD,n_iterations,converged,method,alpha_loss,sd_alpha_loss,alpha_gain,sd_alpha_gain,forget,sd_forget,lambda,sd_lambda,rho_BS,sd_rho_BS,rho_JP,sd_rho_JP,participant
REE_BIASED_SIMPLE,6,180.9510835589848,373.9021671179696,397.8509544006175,-184.67126445012147,0.07499690634746224,77,True,VBMC,-4.656408160851385,0.2673923083250002,4.860978007016057,0.12293596550744358,4.878031122048962,0.10479175314289452,-1.1325083277017796,0.08531360243720701,-7.186933947265772,1.5027923634788773,0.7141639975994144,0.1514709379135597,qfmtmjjy
REE_BIASED_SIMPLE,6,147.05949144670618,306.11898289341235,330.06777017606026,-148.73826304810177,0.06386559058649151,62,True,VBMC,-4.113096512847176,0.5613132993544285,3.989298389857193,0.5494962613099147,4.890228990012427,0.09897683682459851,-1.1382603823574822,0.1939833707629478,-1.9669617128901125,0.24373585373352838,3.31972957100736,0.3759792623118171,l5hj4bqu
1 model n_params negLL AIC BIC ELBO ELBO_SD n_iterations converged method alpha_loss sd_alpha_loss alpha_gain sd_alpha_gain forget sd_forget lambda sd_lambda rho_BS sd_rho_BS rho_JP sd_rho_JP participant
2 REE_BIASED_SIMPLE 6 180.9510835589848 373.9021671179696 397.8509544006175 -184.67126445012147 0.07499690634746224 77 True VBMC -4.656408160851385 0.2673923083250002 4.860978007016057 0.12293596550744358 4.878031122048962 0.10479175314289452 -1.1325083277017796 0.08531360243720701 -7.186933947265772 1.5027923634788773 0.7141639975994144 0.1514709379135597 qfmtmjjy
3 REE_BIASED_SIMPLE 6 147.05949144670618 306.11898289341235 330.06777017606026 -148.73826304810177 0.06386559058649151 62 True VBMC -4.113096512847176 0.5613132993544285 3.989298389857193 0.5494962613099147 4.890228990012427 0.09897683682459851 -1.1382603823574822 0.1939833707629478 -1.9669617128901125 0.24373585373352838 3.31972957100736 0.3759792623118171 l5hj4bqu