update the python script and API

@karlnapf 
merge this first and the dataset https://github.com/shogun-toolbox/shogun-data/pull/49.