Merge pull request #41 from mmschlk/feature-plots

Adds Stacked Bar Plot and tidyies up Network Plot.
mmschlk · Feb 6, 2024 · 321bfcd · 321bfcd
2 parents 20aa04c + c564c36
commit 321bfcd
Show file tree

Hide file tree

Showing 8 changed files with 480 additions and 131 deletions.
diff --git a/docs/source/_static/stacked_bar_exampl.png b/docs/source/_static/stacked_bar_exampl.png
diff --git a/shapiq/__init__.py b/shapiq/__init__.py
@@ -19,7 +19,7 @@
 from .games import DummyGame
 
 # plotting functions
-from .plot import network_plot
+from .plot import network_plot, stacked_bar_plot
 
 # public utils functions
 from .utils import (  # sets.py  # tree.py
@@ -48,6 +48,7 @@
     "DummyGame",
     # plots
     "network_plot",
+    "stacked_bar_plot",
     # public utils
     "powerset",
     "get_explicit_subsets",

diff --git a/shapiq/plot/__init__.py b/shapiq/plot/__init__.py
@@ -1,7 +1,6 @@
 """This module contains all plotting functions for the shapiq package."""
 
 from .network import network_plot
+from .stacked_bar import stacked_bar_plot
 
-__all__ = [
-    "network_plot",
-]
+__all__ = ["network_plot", "stacked_bar_plot"]
diff --git a/shapiq/plot/_config.py b/shapiq/plot/_config.py
@@ -1,12 +1,29 @@
 """This module contains the configuration for the shapiq visualizations."""
 from colour import Color
 
-RED = Color("#ff0d57")
-BLUE = Color("#1e88e5")
-NEUTRAL = Color("#ffffff")
-
 __all__ = [
     "RED",
     "BLUE",
     "NEUTRAL",
+    "LINES",
+    "COLORS_N_SII",
+]
+
+RED = Color("#ff0d57")
+BLUE = Color("#1e88e5")
+NEUTRAL = Color("#ffffff")
+LINES = Color("#cccccc")
+
+COLORS_N_SII = [
+    "#D81B60",
+    "#FFB000",
+    "#1E88E5",
+    "#FE6100",
+    "#7F975F",
+    "#74ced2",
+    "#708090",
+    "#9966CC",
+    "#CCCCCC",
+    "#800080",
 ]
+COLORS_N_SII = COLORS_N_SII * (100 + (len(COLORS_N_SII)))  # repeat the colors list