Is there a way of renaming the output variables when there are multiple variables with the same prefix?
For example if I am inferring the parameter mu, with input shape=3, my pm.summary call for example outputs as expected:
mean sd mc_error hpd_2.5 hpd_97.5 n_eff Rhat
mu_0 -0.19 0.44 0.01 -1.00 0.70 972.71 1.0
mu_1 -0.19 0.40 0.01 -0.93 0.63 1196.89 1.0
mu_2 0.18 0.38 0.01 -0.59 0.90 841.64 1.0
Is there a way of renaming these with custom suffixes from a list of strings e.g.[‘X’,‘Y’,‘Z’] which would output mu_X, mu_Y, mu_Z, as opposed to the default 0,1,2?
Or can you give them new names entirely once the sampling is completed, so when calling pm.summary or pm.traceplot you can map the labels so mu_0 is displayed as X, mu_1 is Y, and mu_2 is Z?