From c38bebf852f7e9e61614ccd863144d83acaee607 Mon Sep 17 00:00:00 2001 From: mei Date: Fri, 19 Feb 2021 19:17:57 +0900 Subject: [PATCH] FIX: the problem of pos_color and negative_color always being reversed. --- fairml/graphing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fairml/graphing.py b/fairml/graphing.py index e20d384..1e02bfc 100644 --- a/fairml/graphing.py +++ b/fairml/graphing.py @@ -61,7 +61,7 @@ def map_x(x): fig = plt.figure(figsize=fig_size) - bar_colors = assign_colors_to_bars(coefficient_values, reverse=True) + bar_colors = assign_colors_to_bars(coefficient_values, reverse=reverse_values) bar_colors = list(np.array(bar_colors)[index_sorted]) plt.barh(pos, sorted_column_values, align='center', color=bar_colors)