Information Gain In Decision Tree Python Code