Skip to content

Add beeswarm plot to SHAP-IQ #406

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Open

Add beeswarm plot to SHAP-IQ #406

wants to merge 11 commits into from

Conversation

annprzy
Copy link

@annprzy annprzy commented Jun 17, 2025

This PR addresses #399, adapting SHAP beeswarm plot to handle interactions and adding it to the plot module.

New features:

  • shapiq/plot/beeswarm.py: adapts SHAP beeswarm plot to handle interactions
  • tests/tests_plots/test_beeswarm.py: adds test for beeswarm plot

Updated:

  • shapiq/__init__.py: adds beeswarm plot to the list of imports and __all__ attribute
  • shapiq/plot/__init__.py: adds beeswarm plot to the list of imports and __all__ attribute

@mmschlk
Copy link
Owner

mmschlk commented Jun 18, 2025

@annprzy, would it technically be possible to work without the skimage import? I see that you only require the conversion of color spaces there. I would love to not blow up the dependencies of shapiq too much and we already use colour or matplotlib which probably can also do what you need to do with skimage. Maybe the code would also work without the conversions even?

Second, I know you asked this in the issue, but if you do have some additional time for this, would you be able to add a quick tutorial for the plot under api_examples like the upset plot? This really does not need to be too elaborate but some quick showcase of the plot?

Thank you for this PR! :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants