trial_index,trial_type,IV_number_choice_only_trials,reward_inference_query,choice_inference_query,reward_prediction_query,choice_prediction_query
0,3,20,0,0,0,0
1,2,20,0,0,0,0
2,3,20,0,0,0,0
3,1,20,0,0,0,0
4,3,20,0,1,0,0
5,2,20,1,0,0,0
6,1,20,0,0,0,0
7,3,20,0,0,1,0
8,2,20,0,0,1,0
9,3,20,0,0,0,1
10,3,20,0,1,0,0
11,3,20,0,0,1,0
12,3,20,0,1,0,0
13,2,20,1,0,0,0
14,1,20,0,0,0,0
15,2,20,0,0,1,1
16,2,20,0,0,0,0
17,3,20,0,0,0,1
18,3,20,0,1,0,0
19,2,20,0,0,0,1
20,2,20,0,0,0,0
21,3,20,0,1,0,0
22,2,20,0,0,0,1
23,2,20,0,0,0,0
24,1,20,0,0,0,0
25,2,20,1,0,0,0
26,1,20,0,0,0,1
27,2,20,0,0,0,0
28,3,20,0,0,0,0
29,2,20,0,0,0,1
30,3,20,0,1,0,0
31,2,20,0,0,0,0
32,2,20,1,0,0,0
33,2,20,1,0,1,0
34,2,20,0,0,0,0
35,3,20,0,0,1,0
36,2,20,1,0,0,0
37,2,20,1,0,0,0
38,2,20,0,0,1,0
39,3,20,0,1,0,0