Fix model selection and learnign rate