diff --git a/cebra/integrations/matplotlib.py b/cebra/integrations/matplotlib.py index 30af7fd4..c2696d4a 100644 --- a/cebra/integrations/matplotlib.py +++ b/cebra/integrations/matplotlib.py @@ -684,7 +684,7 @@ def _to_heatmap_format( else: heatmap_values[i, j] = score_dict[label_i, label_j] - return np.minimum(heatmap_values * 100, 99) + return heatmap_values * 100 def _create_text(self): """Create the text to add in the confusion matrix grid and the title."""