I wrote a class to keep three variables column-name, cut-value and info-gain [1]. I\'m feeding this in a for loop, like this:
def best_split(X, y): best_gain