## Data Collection

This folder contain the code for collect data from LLMs and base line algorithms

*2_arm_baseline_model*: contain base line models (UCB, TS, EG) for the 2 arms task, please only run one sub section under the *get data* section

*4_arm_baseline_model*: contain base line models (UCB, TS, EG) for the 4 arms task, please only run one sub section under the *get data* section

*2_arm_data_collection*: contain code to invoke LLMs for the 2 arm task, please only run one block under each of the choose agent section and choose task section, and makesuer to fill out the api keys with your own api key

*4_arm_data_collection*: contain code to invoke LLMs for the 4 arm task, please only run one block under each of the choose agent section and choose task section, and makesuer to fill out the api keys with your own api key
