Efficient Online Bayesian Inference for Neural Bandits

Installation

pip install fire
pip install ml-collections

There are two ways to reproduce the results from the paper

To reproduce the results, cd into the project folder and run

python bandits test

python bandits run_and_plot

If you only want to reproduce the results, run

python bandits run_experiments

If you have previously reproduced the results and want to reproduce the plots, run

python bandits plot_experiments

The results will be stored inside bandits/figures/.

An alternative way to reproduce the results is to simply open and run subspace_bandits.ipynb