fixes attribute ranking computations

NFFT · Nov 23, 2021 · 015213c · 015213c · mischmi96 · Nov 23, 2021
1 parent 6107e0b
commit 015213c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "ANOVAapprox"
 uuid = "5e027bd6-ab01-4733-8320-e0223e929ebb"
 authors = ["Michael Schmischke <[email protected]> and Felix Bartel <[email protected]>"]
-version = "1.1.2"
+version = "1.1.3"
 
 [deps]
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"

diff --git a/src/analysis.jl b/src/analysis.jl
@@ -66,10 +66,12 @@ function get_AttributeRanking(a::approx, λ::Float64)::Vector{Float64}
 
     for i = 1:length(U)
         u = U[i]
+        weights = 0.0
         for s in u
-            r[s] += gsis[u] * factors[i, s]
+            r[s] += gsis[u] * 1/factors[i, s]
+            weights += 1/factors[i, s]
         end
-        nf += gsis[u] * sum(factors[i, :])
+        nf += weights * gsis[u]
     end
 
     return r ./ nf