Medium Decision Tree Python Sklearn