Add heatmapping preset for CAM methods

Julia-XAI · Nov 17, 2023 · 5658de9 · 5658de9
1 parent 47d4706
commit 5658de9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/explanation.jl b/src/explanation.jl
@@ -6,7 +6,8 @@ Return type of analyzers when calling [`analyze`](@ref).
 * `output`: model output for the given analyzer input
 * `output_selection`: index of the output used for the explanation
 * `analyzer`: symbol corresponding the used analyzer, e.g. `:Gradient` or `:LRP`
-* `heatmap`: symbol indicating a preset heatmapping style, e.g. `:attibution` or `:sensitivity`
+* `heatmap`: symbol indicating a preset heatmapping style,
+    e.g. `:attibution`, `:sensitivity` or `:cam`
 * `extras`: optional named tuple that can be used by analyzers
     to return additional information.
 """

diff --git a/src/heatmaps.jl b/src/heatmaps.jl
@@ -14,6 +14,7 @@ const DEFAULT_HEATMAP_PRESET = HeatmapConfig(
 const HEATMAP_PRESETS = Dict{Symbol,HeatmapConfig}(
     :attribution => HeatmapConfig(:seismic, :sum, :centered),
     :sensitivity => HeatmapConfig(:grays, :norm, :extrema),
+    :cam         => HeatmapConfig(:jet, :sum, :extrema),
 )
 
 # Select HeatmapConfig preset based on heatmapping style in Explanation