Update arviz data extracted from MCMC results #485
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary:
BOLFI and BOLFIRE sample several MCMC chains that could be examined with
arviz
tools, but the currentidata
returned from the sample class does not separate the chains. Here theidata
in BOLFI and BOLFIRE samples is updated to include the individual chains. This means that we can for example usearviz.rhat(idata)
andarviz.plot_trace(idata, kind='rank_bars')
to examine the MCMC results. Meanwhile for examplearviz.plot_posterior(idata)
produces the same result with the current and updatedidata
.Since there is no difference* in how BOLFI and BOLFIRE samples are created based on MCMC chains, this update also combines the
BolfiSample
andBOLFIRESample
classes into anMcmcSample
. In practice this means that we do not need to duplicate code for new functionalities (like theidata
update) in two classes. For example in the current version,plot_traces
is only implemented in theBolfiSample
class while BOLFIRE uses the same MCMC methods to sample the posterior and could benefit from the same analysis.*There was small a difference in that BOLFI uses the keyword
warmup
to save how many samples should be discarded as warmup while BOLFIRE uses the same keyword to save chains where the warmup samples have been discarded. I assumed the latter was a mistake since it would be better that the same keyword is used to save the same information.Please make sure
If your contribution adds, removes or somehow changes the functional behavior of the package, please check that
make lint
,make docs
andmake test
and the proposed changes pass all unit tests (check step 6 of CONTRIBUTING.rst for details)
Copyright and Licensing
Please list the copyright holder for the work you are submitting (this will be you or your assignee, such as a university or company):
By submitting this pull request, the copyright holder is agreeing to license the submitted work under the following licenses: