hierarchical partial pooling

[OriolAbril]

File: https://github.com/pymc-devs/pymc-examples/blob/main/examples/case_studies/hierarchical_partial_pooling.ipynb
Reviewers: @OriolAbril

Changes for discussion

Changes listed in this section are up for discussion, these are ideas on how to improve
the notebook but may not have a clear implementation, or fix some know issue only partially.

ArviZ related

• Consider using new ArviZ labeling capabilities in the plot_forest to avoid having to manually set the yticklabels. We'd instead use coords for that.
  • Note labeller feature in ArviZ has not been released yet, this requires waiting for this.

Notes

Exotic dependencies

None

Computing power

Model runs in roughly a minute

[mjhajharia]

I was working on hierarchical partial pooling last week for a talk proposal!! I'll be happy to take this up, assigning myself if that's okay

[OriolAbril]

That would be great. The only change I had in mind however relies on ArviZ>0.11.2 which I think should be released any day now that gsoc is over? cc @canyon289 was also the release manager here

[canyon289]

@mjhajharia Assigning yourself is great. Well probably release arviz before youre done with your talk which is good. It means when you give your talk people will be able to use the notebook immediately

[azihna]

Working on this with @PrinceAsiedu