"""
Forest Plot
===========
_thumb: .5, .8
"""
import arviz as az
az.style.use('arviz-darkgrid')
trace = az.load_trace('data/centered_eight_trace.gzip')
az.forestplot(trace, varnames=('theta__0', 'theta__1', 'theta__2'))
"""
Joyplot
=======
_thumb: .8, .5
"""
import arviz as az
az.style.use('arviz-darkgrid')
non_centered_data = az.load_data('data/non_centered_eight.nc')
fig, axes = az.forestplot(non_centered_data,
kind='joyplot',
var_names=['theta'],
combined=True,
textsize=11,
joyplot_overlap=3,
colors='white',
r_hat=False,
n_eff=False)
axes[0].set_title('Estimated theta for eight schools model', fontsize=11)
"""
Forest Plot
===========
_thumb: .5, .8
"""
import arviz as az
az.style.use('arviz-darkgrid')
centered_data = az.load_data('data/centered_eight.nc')
non_centered_data = az.load_data('data/non_centered_eight.nc')
fig, axes = az.forestplot([centered_data, non_centered_data],
model_names=['Centered', 'Non Centered'],
var_names=['mu'])
axes[0].set_title('Estimated theta for eight schools model')