Ekeany / Boruta-Shap

A Tree based feature selection tool which combines both the Boruta feature selection algorithm with shapley values.
MIT License
559 stars 86 forks source link

[ENH] #129

Open rwjmiller opened 6 months ago

rwjmiller commented 6 months ago

Current Situation

Boxplots can be generated, but there is no easy way to export them as the plot method doesn't return the figure object that is displayed, and closes it before it can be used for anything else.

Enhancement

Enable saving a plot to a file.

Reasoning

Improved usability

Implementation

Not sure how best to do this between Matplotlib and Seaborn - I have been trying to get it working but no luck so far. Perhaps add an extra parameter to BorutaShap.BorutaShap.plot() where users can specify a path to export the plot as an image file?