It matters little, but a few places in your code has indentation like this. The scikit-learn devs tend to prefer the other line continuation option, e.g. newline between `** 2` and `)`.