Changing the decision function to return n_samples x n_classes shaped decision array