PyMC3: Как использовать pymc3.traceplot() для перекрытия средств на traceplot

Question

Существует функция pymc3.traceplot(), которая отображает трассировочные участки процесса выборки. Я вижу, что функция принимает аргумент lines, который принимает словарь, в котором вы можете передать средства как строковые строки. Как бы вы это сделали?

Answer 1

Вы можете передать любое значение, которое вы хотите не только для среднего.

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

theta это имя переменной в модели и theta_val значение, которое вы хотите построить (перекрытие).

Вы можете вычислить среднее значение от трассы, выполнив:

trace['theta'].mean() 

или вы также можете сделать что-то вроде:

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