rel 0.0.6 - Fallback to StatsFuns for logsumexp()

Author: goedman

notebooks/cars.jl

@@ -80,7 +80,7 @@ end
 
 # ╔═╡ 20ed768a-6008-11eb-13f4-458ca1a29592
 begin
-	lppd = [PSIS.logsumexp(log_lik[:, i] .- log(ns)) for i in 1:n]
+	lppd = [StatsFuns.logsumexp(log_lik[:, i] .- log(ns)) for i in 1:n]
 	pwaic = [var(log_lik[:, i]) for i in 1:n]
 	-2(sum(lppd) - sum(pwaic))
 end

src/logsumexp.jl

@@ -12,6 +12,3 @@ function logsumexp(x::AbstractArray, d::Int64=1)
     end
     result
 end
-
-export
-    logsumexp

src/waic.jl

@@ -8,7 +8,7 @@ function waic( ll::AbstractArray; pointwise=false , log_lik="log_lik" , kwargs..
     pD = zeros(n_obs)
 
     for i in 1:n_obs 
-        lpd[i] = log_sum_exp(ll[:,i]) .- log(n_samples)
+        lpd[i] = logsumexp(ll[:,i]) .- log(n_samples)
         pD[i] = var2(ll[:,i])
     end