david-cortes / contextualbandits

Python implementations of contextual bandits algorithms
http://contextual-bandits.readthedocs.io
BSD 2-Clause "Simplified" License
750 stars 148 forks source link

Add progress bar if tqdm installed #31

Closed alexmill closed 4 years ago

alexmill commented 4 years ago

This commit adds a progress bar to the loop in evaluteRejctionSampling if tqdm installed; otherwise the loop works as before. This isn't the most elegant way to handle this problem, but just a proposal that gets the job done.

david-cortes commented 4 years ago

Thanks, merged. I'll however modify a bit the code as I had made modifications to make it work with more potential inputs for batch sizes.