lazypredict
lazypredict copied to clipboard
Remove "QuantileRegressor" from Supervised.py to avoid system hanging
Removing "QuantileRegressor" which is the cause of the system to hang at 64% or 74%. Tested on multiple size datasets and it solves the problem
This is a simple but very useful PR. @shankarpandala I think you should look into this. It solves a problem many of us are facing. Personally, I manually remove that regressor before using LazyPredict, otherwise, it stalls right after using the Poisson Regressor.
I used the code below to get it running without this fix, but the fix would certainly help a great deal, unless there's some other consideration I do not know of:
import sklearn
from lazypredict import Supervised
Supervised.removed_regressors.append("QuantileRegressor")
Supervised.REGRESSORS.remove(('QuantileRegressor', sklearn.linear_model._quantile.QuantileRegressor))
LazyRegressor = Supervised.LazyRegressor
Then use the LazyRegressor as usual.
Let's rename the PR such it reflects the suggested change...
Changed the title of the MR as suggested.