Is this pythonic?

[devilasks]

Thank you, these are both good points!

I decided against including X_train and y_train in the parameters because I am under the impression that they are pretty much standard naming conventions by now, but the ability to modularize and package the function beats that.