To use this code, download the game Needle Master from the Google Play store. Go into options and enable data collection. Then, plug your Android device into a computer, go to the storage, and copy out the files from the needle_master_trials folder.
Installing the requirements for this python module is as simple as calling python setup.py develop --user
To run DQN on an example environment, call
python3 -m rainbow_dqn.main data/environment_14.txt