# replace with your own
model_dir='/shared/share_mala/implicitbayes/dataset_files/MIND_data/large/N=1000,D=20000,D_eval=10000,method=0111_logistic_withZ_zero_prodsign_last,dimX=5,one_X_per_col=False,flip/bert_Z/models/even_use_Y_0111_20000_1_context_semisynthetic_fixed_bert_0111_logistic_withZ_zero_prodsign_last_sequential_context_dimX=5/fs_sequential:epochs=40,bs=500,lr=0.01,wd=0.01,MLP_layers=3,MLP_width=100,weight_factor=1,max_obs=500,repeat_suffstat=100,Zdim=768,sched=constant,X_MLP_layer=2,X_MLP_width=100,suffstat_eps=1.0,useY,seed=2340923'
usable_Z_file='/shared/share_mala/implicitbayes/dataset_files/MIND_data/large/N=1000,D=20000,D_eval=10000,method=0111_logistic_withZ_zero_prodsign_last,dimX=5,one_X_per_col=False,flip/bert_Z/unused_raw_Z.pt'
generate_Z_file='/shared/share_mala/implicitbayes/dataset_files/MIND_data/large/N=1000,D=20000,D_eval=10000,method=0111_logistic_withZ_zero_prodsign_last,dimX=5,one_X_per_col=False,flip/unused_generate_Z.pt'

dimX=5
T=500
num_arms=10
num_imagined=500
dgp='0111_logistic_withZ_zero_prodsign_last'
finite_horizon_alg=True
bandit_dir='bandit'
verbose=True
finite_horizon_alg=True
alg='sequential_context'
t2p='mlp'
others=' --draw_X_samples True'
for env_idx in {0..500}
do
    sleep .1
    echo $env_idx
    python -m run_bandit_psar --model_dir $model_dir --env_idx $env_idx --T $T --num_arms $num_arms --num_imagined $num_imagined --bandit_alg $alg --dgp $dgp --finite_horizon_alg $finite_horizon_alg --X_dim $dimX --bandit_dir $bandit_dir  --t2p $t2p --verbose $verbose --usable_Z_file $usable_Z_file --generate_Z_file $generate_Z_file $others
done


