trial_index,trial_type,reward_inference_query,choice_inference_query,reward_prediction_query,choice_prediction_query 0,1,0,0,0,0 1,1,0,0,0,0 2,2,0,0,0,0 3,3,0,0,0,0 4,3,0,1,0,0 5,2,0,0,0,0 6,3,0,0,0,1 7,2,0,0,1,0 8,1,0,0,1,0 9,2,1,0,0,0 10,2,0,0,0,0