Just some wrappers around scikit-learn to do more stuffs