Overfitting and underfitting with Python