Update importance.py

vertica · Nov 6, 2024 · 20ce34b · 20ce34b
1 parent 7b4127e
commit 20ce34b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/verticapy/plotting/_plotly/machine_learning/importance.py b/verticapy/plotting/_plotly/machine_learning/importance.py
@@ -73,6 +73,9 @@ def draw(
                 y=self.layout["columns"],
                 orientation="h",
                 name="Postive",
+                marker=dict(
+                    color=self.get_colors()[0],
+                ),
             )
         )
         showlegend = False
@@ -83,6 +86,9 @@ def draw(
                     y=self.layout["columns"],
                     orientation="h",
                     name="Negative",
+                    marker=dict(
+                        color=self.get_colors()[1],
+                    ),
                 )
             )
             showlegend = True