[pymc3][python]pymc3安装后测试代码2

测试环境:

pymc3==3.11.2

代码:

import numpy as np
import pymc3 as pm
import matplotlib.pyplot as plt


if __name__ == '__main__':

    # 生成随机数据
    np.random.seed(123)
    x = np.linspace(0, 1, 100)
    y = 0.5 * x + np.random.normal(0, 0.1, size=100)

    # 定义概率模型
    with pm.Model() as model:
        # 定义先验分布
        alpha = pm.Normal('alpha', mu=0, sd=1)
        beta = pm.Normal('beta', mu=0, sd=1)
        sigma = pm.HalfNormal('sigma', sd=1)

        # 定义线性关系
        mu = alpha + beta * x

        # 定义似然函数
        likelihood = pm.Normal('y', mu=mu, sd=sigma, observed=y)

        # 进行贝叶斯推断
        trace = pm.sample(1000, tune=1000)

    # 可视化结果
    pm.traceplot(trace)
    plt.show()

结果:

相关推荐
Swift社区1 小时前
在 Swift 中实现字符串分割问题:以字典中的单词构造句子
开发语言·ios·swift
没头脑的ht1 小时前
Swift内存访问冲突
开发语言·ios·swift
没头脑的ht1 小时前
Swift闭包的本质
开发语言·ios·swift
wjs20241 小时前
Swift 数组
开发语言
stm 学习ing2 小时前
FPGA 第十讲 避免latch的产生
c语言·开发语言·单片机·嵌入式硬件·fpga开发·fpga
湫ccc3 小时前
《Python基础》之字符串格式化输出
开发语言·python
mqiqe3 小时前
Python MySQL通过Binlog 获取变更记录 恢复数据
开发语言·python·mysql
AttackingLin3 小时前
2024强网杯--babyheap house of apple2解法
linux·开发语言·python