Do you think it would be ok to keep this as a pre-processing step, such as here? As long as we don't have to change anything in the model class, this is basically up to the user (though having some good utils doesn't hurt ;)