de-Rodrigo commited on
Commit
554f351
·
1 Parent(s): 877e0b7
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1536,6 +1536,9 @@ def run_model(model_name):
1536
  heatmap_data = heat_stat.T
1537
  heatmap_data = np.nan_to_num(heat_stat.T, nan=0.0, posinf=0.0, neginf=0.0)
1538
 
 
 
 
1539
  # Color mapper
1540
  if selected_feature in model_options:
1541
  color_mapper = LinearColorMapper(
 
1536
  heatmap_data = heat_stat.T
1537
  heatmap_data = np.nan_to_num(heat_stat.T, nan=0.0, posinf=0.0, neginf=0.0)
1538
 
1539
+ cmap = plt.get_cmap("RdYlGn")
1540
+ red_green_palette = [mcolors.rgb2hex(cmap(i)) for i in np.linspace(0, 1, 256)]
1541
+
1542
  # Color mapper
1543
  if selected_feature in model_options:
1544
  color_mapper = LinearColorMapper(