A class for using ML algorithms for spike prediction - View it on GitHub