information_gain function

Information gain