Information Gain Python