Diagnosing divergences when clipping is involved

This notebooks is unfortunately quite out of date. You can make this plot in one line using arviz.plot_pair:

import arviz as az
az.plot_pair(short_trace, 
             var_names=['theta', 'mu'],
             coords={'theta_dim_0':[0]}, 
             divergences=True)

The readability can be improved by adding coords to the model, as shown in this notebook.

1 Like