arviz-devs / arviz

Exploratory analysis of Bayesian models with Python
https://python.arviz.org
Apache License 2.0
1.59k stars 393 forks source link

Add R* convergence diagnostic #1367

Open avehtari opened 4 years ago

avehtari commented 4 years ago

Add R* diagnostic as described in

The reference R code available at

ahartikainen commented 3 years ago

Adding some initial tests here

https://gist.github.com/ahartikainen/b892b487df9c067d8e91cdd4d986d3af

I'm not sure what library we should use. And if we select some, then I think it should be "optional".

I think we should also do a comparison test between different libs to see how they perform (time vs accuracy).

sethaxen commented 3 years ago

In case it's useful, here's an example calling MCMCChains.jl's implementation of R* on the same data: https://gist.github.com/sethaxen/b22d9a9ca70530d8f0bce1687a09633f.