Public code by Frida Heskebeck from her research on Multi-armed bandits for Brain-Computer Interfaces.