adjust the API of KL methods for model selection

make the API work for `GradientModelSelection`
@karlnapf 
please merge this.
Please let me know if you have any question about this.
If not, please merge it so that I can send the nb.