Remove unnecessary module loads. Get all relevant classifiers and reg…

shankarpandala / lazypredict

Lazy Predict help build a lot of basic models without much code and helps understand which models works better without any parameter tuning

MIT License

2.78k stars 321 forks source link

Closed sanromd closed 3 years ago

sanromd commented 3 years ago

Modifies Supervised.py to remove the ModuleNotFoundError during load. The changes makes the code compatible with sklearn 0.23.x and 0.24.x.

Summary of changes

remove import of sklearn since the calls to the regressors and classifiers are in CLASSIFIERS and REGRESSORS
simplify removal of unused classifiers and regressors during creation (list comprehension)

shankarpandala commented 3 years ago

@sanromd Thankyou for your contribution!