Bayesian Bandit in Python