fix interface

xukai92 · Sep 11, 2019 · 3fef39c · 3fef39c
1 parent 35a1c63
commit 3fef39c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/ratio/moment_matching.jl b/src/ratio/moment_matching.jl
@@ -26,14 +26,16 @@ gaussian_gram_by_pairwise_dot(pdot; σ=1) = exp.(-pdot ./ 2(σ ^ 2))
 gaussian_gram(x; σ=1) = gaussian_gram_by_pairwise_dot(pairwise_dot(x); σ=σ)
 gaussian_gram(x, y; σ=1) = gaussian_gram_by_pairwise_dot(pairwise_dot(x, y); σ=σ)
 
-function estimate_r_de(x_de, x_nu, get_r_hat=get_r_hat_numerically; σs=[sqrt(median([pdot_dede..., pdot_denu...]))])
+function estimate_r_de(x_de, x_nu; get_r_hat=get_r_hat_analytical, σs=nothing, kwargs...)
     pdot_dede = pairwise_dot_kai(x_de)
     pdot_denu = pairwise_dot_kai(x_de, x_nu)
 
+    if isnothing(σs); σs = [sqrt(median([pdot_dede..., pdot_denu...]))]; end
+
     Kdede = mean([gaussian_gram_by_pairwise_dot(pdot_dede; σ=σ) for σ in σs])
     Kdenu = mean([gaussian_gram_by_pairwise_dot(pdot_denu; σ=σ) for σ in σs])
 
-    return get_r_hat(Kdede, Kdenu)
+    return get_r_hat(Kdede, Kdenu; kwargs...)
 end
 
 function get_r_hat_numerically(Kdede, Kdenu; positive=true, normalisation=true)