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

结果:

相关推荐
疯一样的码农3 分钟前
Python 正则表达式(RegEx)
开发语言·python·正则表达式
&岁月不待人&25 分钟前
Kotlin by lazy和lateinit的使用及区别
android·开发语言·kotlin
StayInLove29 分钟前
G1垃圾回收器日志详解
java·开发语言
无尽的大道37 分钟前
Java字符串深度解析:String的实现、常量池与性能优化
java·开发语言·性能优化
爱吃生蚝的于勒40 分钟前
深入学习指针(5)!!!!!!!!!!!!!!!
c语言·开发语言·数据结构·学习·计算机网络·算法
binishuaio1 小时前
Java 第11天 (git版本控制器基础用法)
java·开发语言·git
zz.YE1 小时前
【Java SE】StringBuffer
java·开发语言
就是有点傻1 小时前
WPF中的依赖属性
开发语言·wpf
洋2401 小时前
C语言常用标准库函数
c语言·开发语言
进击的六角龙1 小时前
Python中处理Excel的基本概念(如工作簿、工作表等)
开发语言·python·excel