Python Supervised Binary Decision Tree