fix issue #274

Author: paulbkoch

python/interpret-core/interpret/data/_response.py

@@ -449,14 +449,13 @@ def visualize(self, key=None):
         if key is None:
             y = data_dict["y"]
             values, counts = np.unique(y, return_counts=True)
-            trace1 = go.Bar(x=[values[0]], y=[counts[0]])
-            trace2 = go.Bar(x=[values[1]], y=[counts[1]])
+            traces = [go.Bar(x=[values[i]], y=[counts[i]]) for i in range(len(values))]
             layout = go.Layout(
                 title="Response Distribution",
                 showlegend=False,
                 xaxis={"type": "category"},
             )
-            return go.Figure(data=[trace1, trace2], layout=layout)
+            return go.Figure(data=traces, layout=layout)
 
         # Show feature graph
         x = data_dict["x"]
@@ -481,7 +480,7 @@ def visualize(self, key=None):
             data_dict = {"data_type": "univariate", "names": values, "scores": counts}
             fig = plot_density(
                 data_dict,
-                color=COLORS[idx],
+                color=COLORS[idx % len(COLORS)],
                 name=str(current_class),
                 is_categorical=is_categorical,
             )

python/interpret-core/interpret/visual/plot.py

@@ -13,7 +13,19 @@
 
 _log = logging.getLogger(__name__)
 
-COLORS = ["#1f77b4", "#ff7f0e", "#808080"]
+COLORS = [
+    "#1f77b4",
+    "#ff7f0e",
+    "#808080",
+    "#2ca02c",
+    "#d62728",
+    "#9467bd",
+    "#8c564b",
+    "#e377c2",
+    "#7f7f7f",
+    "#bcbd22",
+    "#17becf",
+]
 
 
 def is_multiclass_global_data_dict(data_dict):