How to avoid memory overloads using SciKit Learn