If you need the class probabilities, then you can apply a Softmax function to the model outputs. The binary predictions are obtained by applying
np.argmax() to the model outputs. Using a
softmax instead of the
argmax function would give the probabilities.
This code is no longer maintained. Please check SimpleTransformers for the updated version.