make it easier to use scikit-learn for classification