Restructure src folder

Create `src/lrp` subfolder

Author: adrhill

src/ExplainableAI.jl

@@ -21,16 +21,16 @@ include("compat.jl")
 include("neuron_selection.jl")
 include("analyze_api.jl")
 include("types.jl")
-include("flux.jl")
+include("flux_utils.jl")
 include("utils.jl")
-include("canonize.jl")
 include("input_augmentation.jl")
 include("gradient.jl")
-include("lrp_checks.jl")
-include("lrp_rules.jl")
-include("lrp.jl")
+include("lrp/canonize.jl")
+include("lrp/checks.jl")
+include("lrp/rules.jl")
+include("lrp/lrp.jl")
 include("heatmap.jl")
-include("imagenet.jl")
+include("preprocessing.jl")
 export analyze
 
 # Analyzers

src/flux.jl renamed to src/flux_utils.jl

@@ -107,3 +107,30 @@ function check_model(::Val{:LRP}, c::Chain; verbose=true)
         throw(ArgumentError("Unknown or unsupported activation functions found in model"))
     end
 end
+
+# Utils for printing model check summary using PrettyTable.jl
+_print_name(layer) = "$layer"
+_print_name(layer::Parallel) = "Parallel(...)"
+_print_activation(layer) = hasproperty(layer, :σ) ? "$(layer.σ)" : "—"
+_print_activation(layer::Parallel) = "—"
+
+function _show_check_summary(
+    c::Chain, layer_names, layer_checks, activation_names, activation_checks
+)
+    hl_pass = Highlighter((data, i, j) -> j in (3, 5) && data[i, j]; foreground=:green)
+    hl_fail = Highlighter((data, i, j) -> j in (3, 5) && !data[i, j]; foreground=:red)
+    data = hcat(
+        collect(1:length(c)),
+        layer_names,
+        collect(layer_checks),
+        activation_names,
+        collect(activation_checks),
+    )
+    pretty_table(
+        data;
+        header=["", "Layer", "Layer supported", "Activation", "Act. supported"],
+        alignment=[:r, :l, :r, :c, :r],
+        highlighters=(hl_pass, hl_fail),
+    )
+    return nothing
+end

src/canonize.jl renamed to src/lrp/canonize.jl

@@ -76,39 +76,14 @@ ones_like(x::AbstractArray) = ones(eltype(x), size(x))
 ones_like(x::Number) = oneunit(x)
 
 function keep_positive!(x::AbstractArray{T}) where {T}
-    x[x .< 0] .= zero(T)
+    z = zero(T)
+    x[x .< 0] .= z
     return x
 end
 function keep_negative!(x::AbstractArray{T}) where {T}
-    x[x .> 0] .= zero(T)
+    z = zero(T)
+    x[x .> 0] .= z
     return x
 end
 keep_positive(x) = keep_positive!(deepcopy(x))
 keep_negative(x) = keep_negative!(deepcopy(x))
-
-# Utils for printing model check summary using PrettyTable.jl
-_print_name(layer) = "$layer"
-_print_name(layer::Parallel) = "Parallel(...)"
-_print_activation(layer) = hasproperty(layer, :σ) ? "$(layer.σ)" : "—"
-_print_activation(layer::Parallel) = "—"
-
-function _show_check_summary(
-    c::Chain, layer_names, layer_checks, activation_names, activation_checks
-)
-    hl_pass = Highlighter((data, i, j) -> j in (3, 5) && data[i, j]; foreground=:green)
-    hl_fail = Highlighter((data, i, j) -> j in (3, 5) && !data[i, j]; foreground=:red)
-    data = hcat(
-        collect(1:length(c)),
-        layer_names,
-        collect(layer_checks),
-        activation_names,
-        collect(activation_checks),
-    )
-    pretty_table(
-        data;
-        header=["", "Layer", "Layer supported", "Activation", "Act. supported"],
-        alignment=[:r, :l, :r, :c, :r],
-        highlighters=(hl_pass, hl_fail),
-    )
-    return nothing
-end