File tree Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Original file line number Diff line number Diff line change @@ -590,6 +590,25 @@ class chains {
590
590
double split_effective_sample_size (const std::string& name) const {
591
591
return split_effective_sample_size (index (name));
592
592
}
593
+
594
+ double split_potential_scale_reduction (const int index) const {
595
+ int n_chains = num_chains ();
596
+ std::vector<const double *> draws (n_chains);
597
+ std::vector<size_t > sizes (n_chains);
598
+ int n_kept_samples = 0 ;
599
+ for (int chain = 0 ; chain < n_chains; ++chain) {
600
+ n_kept_samples = num_kept_samples (chain);
601
+ draws[chain]
602
+ = samples_ (chain).col (index).bottomRows (n_kept_samples).data ();
603
+ sizes[chain] = n_kept_samples;
604
+ }
605
+
606
+ return analyze::compute_split_potential_scale_reduction (draws, sizes);
607
+ }
608
+
609
+ double split_potential_scale_reduction (const std::string& name) const {
610
+ return split_potential_scale_reduction (index (name));
611
+ }
593
612
};
594
613
595
614
} // namespace mcmc
You can’t perform that action at this time.
0 commit comments