Scikit-Learn is a super useful machine learning library for Python.
I use this pattern all the time. It's super useful both for hyperparameter tuning, but also for quick checks as to whether you really need to bother building a more complex model.