Thompson Sampling - How to get the dataset