Instead of keeping all of the modules in the same utils.py file, break them out into separate submodules. This will help prevent situations where, for example, a user imports matplotlib when they're only using the n_way_models function (that doesn't use matplotlib).