标签: python bayesian pymc3
有一个函数pymc3.traceplot()可以绘制采样过程的跟踪图。我看到该函数接受一个带有字典的参数lines,您可以在其中传递作为要绘制的线的方法。你会怎么做呢?
pymc3.traceplot()
lines
答案 0 :(得分:1)
您可以传递任何您想要的值,而不仅仅是平均值。
theta_val = 0.35 pm.traceplot(trace, lines={'theta':theta_val})
theta是模型中变量的名称,theta_val是您要绘制的值(重叠)。
theta
theta_val
您可以通过执行以下操作来计算跟踪的平均值:
trace['theta'].mean()
或者你也可以这样做:
lines = {var:trace[var].mean() for var in trace.varnames}