Merge pull request #39 from flatironinstitute/drghmc

verified changes for DR-G-HMC sampler
flatironinstitute · Oct 30, 2023 · deb13a6 · deb13a6
2 parents d32dcb5 + 403c8ca
commit deb13a6
Show file tree

Hide file tree

Showing 3 changed files with 819 additions and 0 deletions.
diff --git a/bayes_kit/__init__.py b/bayes_kit/__init__.py
@@ -2,6 +2,7 @@
 from .autocorr import autocorr
 
 # classes
+from .drghmc import DrGhmcDiag
 from .ensemble import Stretcher
 from .ess import ess, ess_imse, ess_ipse
 from .hmc import HMCDiag
@@ -12,11 +13,13 @@
 from .smc import TemperedLikelihoodSMC
 
 __all__ = [
+    "DrGhmcDiag",
     "HMCDiag",
     "MALA",
     "Metropolis",
     "MetropolisHastings",
     "TemperedLikelihoodSMC",
+    "Stretcher",
     "ess",
     "ess_imse",
     "ess_ipse",