Decision Tree Data Modeling Python