arviz-devs / arviz-plots

ArviZ modular plotting
https://arviz-plots.readthedocs.io
Apache License 2.0
2 stars 1 forks source link

add plot_trace #1

Closed aloctavodia closed 11 months ago

aloctavodia commented 11 months ago

Closes #13

This follows the decision that plot_trace will be a plot without the densities (kde or histograms).

I started with this example because it seems simple enough to get familiar with the code. Please let me know what I am doing wrong.

Also not sure what should I do to add divergences, given that divergences are defined in a separate group.

This is how the default plots look like for the centered_eight dataset

default

and this is with some customization

plot_trace(post, var_names=["mu", "tau"], sample_dims=["draw"], pc_kwargs={"plot_grid_kws": {"figsize": (12, 7), "sharey":"row"}, "col_wrap":4})

customized


:books: Documentation preview :books:: https://arviz-plots--1.org.readthedocs.build/en/1/

codecov-commenter commented 11 months ago

Welcome to Codecov :tada:

Once merged to your default branch, Codecov will compare your coverage reports and display the results in this comment.

Thanks for integrating Codecov - We've got you covered :open_umbrella:

OriolAbril commented 11 months ago

Also not sure what should I do to add divergences, given that divergences are defined in a separate group.

There is no clear way to go about that yet. I think the clearest way to go about it is allowing to define different data when looping/facetting than when plotting

aloctavodia commented 11 months ago

Is this ok to merge? Or what else should I do?