m0_56259459
2022-03-11 17:06
采纳率: 100%
浏览 18

如何使用pymc3时,pm.traceplot(trace)出现问题?(语言-python)

问题遇到的现象和发生背景
问题相关代码,请勿粘贴截图

import numpy as np
import pymc3 as pm
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from scipy import optimize
import sys

def main_1():
# 设置随机数种子
np.random.seed(123)

alpha = 1
sigma = 1
beta = [1, 2.5]

N = 100

X1 = np.random.randn(N)
X2 = np.random.randn(N)

Y = alpha + beta[0] * X1 + beta[1] * X2 + np.random.randn(N) * sigma



basic_model = pm.Model()
with basic_model:
    alpha = pm.Normal('alpha', mu=0, sd=10)
    beta = pm.Normal('beta', mu=0, sd=10, shape=2)
    sigma = pm.HalfNormal('sigma', sd=1)

    mu = alpha + beta[0] * X1 + beta[1] * X2

    Y_obs = pm.Normal('Y_obs', mu=mu, sd=sigma, observed=Y)

with basic_model:
    # 用MAP获得初始点
    start = pm.find_MAP(method='BFGS')

    # 实例化采样器
    step = pm.Slice(vars=[sigma])

    # 对后验分布进行5000次采样
    trace = pm.sample(5000, step=step, start=start)

pm.traceplot(trace)

if name == 'main':
sys.exit(main_1())

运行结果及报错内容

img

我的解答思路和尝试过的方法
我想要达到的结果
  • 写回答
  • 好问题 提建议
  • 追加酬金
  • 关注问题
  • 邀请回答

2条回答 默认 最新

相关推荐 更多相似问题