python-2.7 - 在 PyMC3 中保存跟踪图中的数据
问题描述
下面是简单贝叶斯线性回归的代码。在我获得参数的轨迹和图之后,有什么方法可以将创建图的数据保存在文件中,这样如果我需要再次绘制它,我可以简单地从文件中的数据中绘制它而不是再次运行整个模拟?
import pymc3 as pm
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0,9,5)
y = 2*x + 5
yerr=np.random.rand(len(x))
def soln(x, p1, p2):
return p1+p2*x
with pm.Model() as model:
# Define priors
intercept = pm.Normal('Intercept', 15, sd=5)
slope = pm.Normal('Slope', 20, sd=5)
# Model solution
sol = soln(x, intercept, slope)
# Define likelihood
likelihood = pm.Normal('Y', mu=sol,
sd=yerr, observed=y)
# Sampling
trace = pm.sample(1000, nchains = 1)
pm.traceplot(trace)
print pm.summary(trace, ['Slope'])
print pm.summary(trace, ['Intercept'])
plt.show()
解决方案
有两种简单的方法可以做到这一点:
使用 3.4.1 之后的版本(目前这意味着从 master 安装,带有
pip install git+https://github.com/pymc-devs/pymc3
)。有一个新功能可以有效地保存和加载轨迹。请注意,您需要访问创建跟踪的模型:... pm.save_trace(trace, 'linreg.trace') # later with model: trace = pm.load_trace('linreg.trace')
使用
cPickle
(或pickle
在 python 3 中)。请注意,这pickle
至少有点不安全,不要从不受信任的来源解开数据:import cPickle as pickle # just `import pickle` on python 3 ... with open('trace.pkl', 'wb') as buff: pickle.dump(trace, buff) #later with open('trace.pkl', 'rb') as buff: trace = pickle.load(buff)
推荐阅读
- java - Java 列表
来自 JPA 查询 - awk - awk - 从文件中查找数字
- bash - 将文件名添加为 .txt 文件中的列并连接到单个文件中
- javascript - 为 livereload 编译 gulpfile 时出错
- python - ValueError:所有参数的形状
- javascript - 我的网页似乎可以在某些平台上运行,而不能在其他平台上运行,css 没有得到应用,这是为什么呢?
- c# - 将 AbsoluteLayout 用于背景图像,但某些元素未正确定位 Xamarin Forms
- angular - Angular6 - GET net::ERR_CONNECTION_REFUSED 的 HttpClient 错误处理
- unity3d - 如何在 Unity 项目之间共享新的 SteamVR 动作集和绑定
- debugging - 为什么tomcat在使用类似于“something.com-xyz”的url时返回“400 Bad Request”