You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
It seems like the eval_metric you can specify during initiation of the EarlyStoppingShapRFECV class is not being used in the case of a LGBMClassifier (did not test the other tree-based methods).
Environment (please complete the following information):
(This is just the exact example data set + call to EarlyStoppingShapRFECV from here just with verbose=2 added and actually using the correct model (model=model))
I would expect the parameter for eval_metric to be used during the evaluations for the early-stopping.
Potential solution
I think the issue comes from the fact that for LGBMClassifier the eval_metric is being set here but this does not work as the class does not have this parameter. What should be done (I assume) is this:
Describe the bug
It seems like the
eval_metric
you can specify during initiation of theEarlyStoppingShapRFECV
class is not being used in the case of aLGBMClassifier
(did not test the other tree-based methods).Environment (please complete the following information):
To Reproduce
(This is just the exact example data set + call to
EarlyStoppingShapRFECV
from here just withverbose=2
added and actually using the correct model (model=model
))Error traceback
There is no error but with
verbose=2
you can see:Expected behavior
I would expect the parameter for
eval_metric
to be used during the evaluations for the early-stopping.Potential solution
I think the issue comes from the fact that for
LGBMClassifier
theeval_metric
is being set here but this does not work as the class does not have this parameter. What should be done (I assume) is this:eval_metric=self.eval_metric
to this asLGBMClassifier
's fit method does have aneval_metric
parameterThe text was updated successfully, but these errors were encountered: