Move expansion of probabilities to specialized sklearn Model/Fitter classes.