# Data

This folder contains data we collected from LLM and base line algorithms, it contain following folders:

## 2armsdata

Contain data of LLMs and the base line algorithms performing the 2-arms bandit task in csv format

### Data Format:

subject: the subject id, range from 0-14

block: the block id, range from 1-20

trial: the trial index, range from 1-10

mu1: the mean used to generate the reward for the first arm

mu2: the mean used to generate the reward for the second arm

choice: the arm the agent choosed

reward: the reward generated for the choosen arm

expleanion: if the agent is a LLM and asked to explain, the outputed expleanion will be recorded in this colum, other wise this colum will be blank

## 4 arms bandit task data

This folder contain data of LLMs and the base line algorithm performing the 4-arms bandit task in numpy array format

### Data Format:

Each array is in the shape of (1,num_Subject,num_Trail,6), where the last axis contain the following data:
index 0: the choice the subject made in that trail with 1-based index

index 1: the reward given for the subject in that trail

index 2: the reward for arm 1 at that trail in the templet for the subject

index 3: the reward for arm 2 at that trail in the templet for the subject

index 4: the reward for arm 3 at that trail in the templet for the subject

index 5: the reward for arm 4 at that trail in the templet for the subject
