[scikit-learn] Issue with DecisionTreeClassifier