adjust the API for model selection

make the API work for `GradientModelSelection`