model_selection function

Visualising model selection