Apparently there are no built-in arguments that you can give to traceplot to change the displayed labels from the variable’s name to something else.
Maybe @colcarroll knows a way, possibly using arviz.
If in the end there are no built-in ways, you can either
- Get a dataframe from the trace, change the indexes names as you like, and then hand it over to
arviz.plot_trace - Add deterministics to your model with your desired name.