Unlike RStan, PyStan seems to overlapping all traceplot chains with the same color. Is it possible to distinguish the different chains when using the traceplot function?
Have you tried arviz traceplot functionality
import arviz as az ... az.plot_trace(fit)
Thank you for your email. I appreciate it. Unfortunately, that doesn’t separate the chains too. I have 8 chains but I can’t view them separately. I think this is a shortcoming of PyStan compared to RStan.
PyStan doesn’t have plotting (2.x had, 3.0 does not)
(legend is optional)
az.plot_trace(fit, compact=False, legend=True)
As @ahartikainen mentioned, setting
compact=False defaults to using color to distinguish chains, whereas
compact=True (default value from
compact) uses color for different dimensions of multidim variables and linestyle for different chains. But both those defaults can be set manually via
Perfect! Thanks a lot.