0b9e1f064ab1ce1d994f86686e7d662a46095e36,doc/source/notebooks/advanced/mcmc.pct.py,,marginal_samples,#Any#Any#Any#,198

Before Change

def marginal_samples(samples, parameters_dict, y_axis_label):
    param_to_name = {param: name for name, param in parameters_dict.items()}
    fig, axarr = plt.subplots(1, len(param_to_name), figsize=(15, 3), constrained_layout=True)
    for i, param in enumerate(model.trainable_parameters):
        ax = axarr[i]
        ax.hist(np.stack(samples[i]).reshape(-1, 1), bins=20)
        ax.set_title(param_to_name[param])
    fig.suptitle(y_axis_label)
    plt.show()

After Change

// %%
def marginal_samples(samples, parameters, y_axis_label):
    fig, axes = plt.subplots(1, len(param_to_name), figsize=(15, 3), constrained_layout=True)
    for ax, val, param in zip(axes, samples, parameters):
        ax.hist(np.stack(val).flatten(), bins=20)
        ax.set_title(param_to_name[param])
    fig.suptitle(y_axis_label)
    plt.show()

Instances