I have a dataset with imbalanced classes that i just resampled to even the classes, my problem is that i wish to return an array with all probabilities of the test samples b