From 20ce34b2bb6767948d4017db508fc3db66d0ef03 Mon Sep 17 00:00:00 2001 From: Badr Date: Wed, 6 Nov 2024 07:13:57 -0500 Subject: [PATCH] Update importance.py --- verticapy/plotting/_plotly/machine_learning/importance.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/verticapy/plotting/_plotly/machine_learning/importance.py b/verticapy/plotting/_plotly/machine_learning/importance.py index 30c33a828..e8eb1f0bc 100644 --- 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