Tweak model selection to select learning rate too