Change default HMC Metric for HamiltonianMC MCMCProposal to UnitEucli…

…deanMetric
bat · Feb 13, 2025 · 03971c1 · 03971c1
1 parent 0cb9e9a
commit 03971c1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/ext/ahmc_impl/ahmc_sampler_impl.jl b/ext/ahmc_impl/ahmc_sampler_impl.jl
@@ -133,7 +133,6 @@ function BAT.mcmc_propose!!(mc_state::HMCState)
 end
 
 function BAT._accept_reject!(mc_state::HMCState, accepted::Bool, p_accept::Float64, current::Integer, proposed::Integer)
-    # @unpack samples, proposal = mc_state
     samples = mc_state.samples
     proposal = mc_state.proposal
 

diff --git a/src/extdefs/ahmc_defs/ahmc_alg.jl b/src/extdefs/ahmc_defs/ahmc_alg.jl
@@ -35,7 +35,7 @@ $(TYPEDFIELDS)
     IT,
     TC
 } <: MCMCProposal
-    metric::MT = DiagEuclideanMetric()
+    metric::MT = UnitEuclideanMetric()
     integrator::IT = ext_default(pkgext(Val(:AdvancedHMC)), Val(:DEFAULT_INTEGRATOR))
     termination::TC = ext_default(pkgext(Val(:AdvancedHMC)), Val(:DEFAULT_TERMINATION_CRITERION))
 end