I think we should have multiple strategies here.
Could you add a mechanism for specifying a multi-class probabilities method?
An enum should do it, can be set via a method and the rescale method checks which method to implement and gives an error message if not supported.
What do you think of this?