mlxtend
mlxtend copied to clipboard
Published
20 hours ago •
rasbt
rasbt
create_counterfactual enhancements (minimization method and fixed features)
Discussed in https://github.com/rasbt/mlxtend/discussions/1028
Originally posted by farbodr April 17, 2023 I have a couple of questions about create_counterfactual?
- On a small table with the shape of (461, 75) it took about 22 min to return one row. Is there any way to speed this up?
- Is there any way to limit the features it uses? In other words, can I freeze some features and generate counterfactuals only for some of the features? For instance, in a healthcare use case, we know certain features can not be changed, like a health condition, but there are other features that can be changed that can impact the prediction.
Thanks, FR
Definitely good points.
-
Off the top off my head, changing the minimization method might help, right now it's hardcoded to be Nelder-Mead. (https://github.com/rasbt/mlxtend/blob/master/mlxtend/evaluate/counterfactual.py#L113)
-
And keeping some features fixed would probably help, too.
These would both require introducing 2 additional function arguments, which shouldn't be too complicated. But I am not sure if I have the capacity in the near future. Would appreciate PRs on that though!
PS: Converted this to an issue because these are good enhancement ideas!