[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()

结果:

相关推荐
点云SLAM16 分钟前
PyTorch 中.backward() 详解使用
人工智能·pytorch·python·深度学习·算法·机器学习·机器人
B1118521Y461 小时前
flask的使用
后端·python·flask
悟能不能悟1 小时前
js闭包问题
开发语言·前端·javascript
潼心1412o1 小时前
C语言(长期更新)第15讲 指针详解(五):习题实战
c语言·开发语言
Learn Beyond Limits1 小时前
Transfer Learning|迁移学习
人工智能·python·深度学习·神经网络·机器学习·ai·吴恩达
Murphy_lx2 小时前
Lambda表达式
开发语言·c++
yangpipi-2 小时前
C++并发编程-23. 线程间切分任务的方法
开发语言·c++
love530love3 小时前
【保姆级教程】阿里 Wan2.1-T2V-14B 模型本地部署全流程:从环境配置到视频生成(附避坑指南)
人工智能·windows·python·开源·大模型·github·音视频