Add printout of burn-in period to lfmcmc print (#39)

UofUEpiBio · Dec 9, 2024 · cfbc814 · cfbc814
1 parent 2f42c2e
commit cfbc814
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/epiworld.hpp b/epiworld.hpp
@@ -2181,7 +2181,8 @@ inline void LFMCMC<TData>::print(size_t burnin) const
     printf_epiworld("___________________________________________\n\n");
     printf_epiworld("LIKELIHOOD-FREE MARKOV CHAIN MONTE CARLO\n\n");
 
-    printf_epiworld("N Samples : %zu\n", m_n_samples);
+    printf_epiworld("N Samples (total) : %zu\n", m_n_samples);
+    printf_epiworld("N Samples (after burn-in period) : %zu\n", m_n_samples - burnin);
 
     std::string abbr;
     epiworld_double elapsed;

diff --git a/include/epiworld/math/lfmcmc/lfmcmc-meat-print.hpp b/include/epiworld/math/lfmcmc/lfmcmc-meat-print.hpp
@@ -73,7 +73,8 @@ inline void LFMCMC<TData>::print(size_t burnin) const
     printf_epiworld("___________________________________________\n\n");
     printf_epiworld("LIKELIHOOD-FREE MARKOV CHAIN MONTE CARLO\n\n");
 
-    printf_epiworld("N Samples : %zu\n", m_n_samples);
+    printf_epiworld("N Samples (total) : %zu\n", m_n_samples);
+    printf_epiworld("N Samples (after burn-in period) : %zu\n", m_n_samples - burnin);
 
     std::string abbr;
     epiworld_double elapsed;