PyMC3:如何使用pymc3.traceplot()重叠意味着traceplot

时间:2017-02-08 15:12:27

标签: python bayesian pymc3

有一个函数pymc3.traceplot()可以绘制采样过程的跟踪图。我看到该函数接受一个带有字典的参数lines,您可以在其中传递作为要绘制的线的方法。你会怎么做呢?

1 个答案:

答案 0 :(得分:1)

您可以传递任何您想要的值,而不仅仅是平均值。

theta_val = 0.35
pm.traceplot(trace, lines={'theta':theta_val})

enter image description here

theta是模型中变量的名称,theta_val是您要绘制的值(重叠)。

您可以通过执行以下操作来计算跟踪的平均值:

trace['theta'].mean()

或者你也可以这样做:

lines = {var:trace[var].mean() for var in trace.varnames}