Here are a few lines to add the Gittins Index to simulations if we want to - it requires a csv data file with the indices which I have but need to share

[sofiasvillar]

ADAGIO/evaluation_functions.R

Line 243 in 75f4021

else if(adaptation%in%c(‘GI’)){
mydata = read.csv("Gittins.csv")

if((successes[2]+1 > 749) | (failures[2]+1 > 749)) GI0 <- (successes[2]+1)/(successes[2]+2+fails[2])
if((successes[1]+1 > 749) | (failures[1]+1 > 749)) GI1 <- (successes[1]+1)/(successes[1]+2+fails[2])

if((successes[2]+1 <= 749) & (failures[2]+1 <= 749)) GI0 <- mydata[successes[2]+1,1+fails[2]]
if((successes[1]+1 <= 749) & (failures[1]+1 <= 749)) GI1 <- mydata[successes[1]+1,1+fails[1]]

if(GI1==0) GI1 <- (successes[1]+1)/(successes[1]+2+fails[1])
if(GI0==0) GI0 <- (successes[2]+1)/(successes[2]+2+fails[2])
if(GI1==0) GI1 <- (successes[1]+1)/(successes[1]+2+fails[1])

if(GI0< GI1) allocation_rate <- 1
if(GI0> GI1) allocation_rate <- 0
if(GI0==GI1) allocation_rate <- 0.5
}