@@ -17,26 +17,25 @@ Visualize explanation.
17
17
Assumes the Flux's WHCN convention (width, height, color channels, batch size).
18
18
19
19
## Keyword arguments
20
- -`cs::ColorScheme`: ColorScheme that is applied.
20
+ - `cs::ColorScheme`: ColorScheme that is applied.
21
21
When calling `heatmap` with an `Explanation` or analyzer, the method default is selected.
22
22
When calling `heatmap` with an array, the default is `ColorSchemes.bwr`.
23
- -`reduce::Symbol`: How the color channels are reduced to a single number to apply a colorscheme.
23
+ - `reduce::Symbol`: How the color channels are reduced to a single number to apply a colorscheme.
24
24
The following methods can be selected, which are then applied over the color channels
25
25
for each "pixel" in the attribution:
26
26
- `:sum`: sum up color channels
27
27
- `:norm`: compute 2-norm over the color channels
28
28
- `:maxabs`: compute `maximum(abs, x)` over the color channels in
29
29
When calling `heatmap` with an `Explanation` or analyzer, the method default is selected.
30
30
When calling `heatmap` with an array, the default is `:sum`.
31
- -`normalize::Symbol`: How the color channel reduced heatmap is normalized before the colorscheme is applied.
31
+ - `normalize::Symbol`: How the color channel reduced heatmap is normalized before the colorscheme is applied.
32
32
Can be either `:extrema` or `:centered`.
33
33
When calling `heatmap` with an `Explanation` or analyzer, the method default is selected.
34
34
When calling `heatmap` with an array, the default for use with the `bwr` colorscheme is `:centered`.
35
- -`permute::Bool`: Whether to flip W&H input channels. Default is `true`.
35
+ - `permute::Bool`: Whether to flip W&H input channels. Default is `true`.
36
36
37
37
**Note:** these keyword arguments can't be used when calling `heatmap` with an analyzer.
38
38
"""
39
-
40
39
function heatmap (
41
40
attr:: AbstractArray ;
42
41
cs:: ColorScheme = ColorSchemes. bwr,
0 commit comments