0

运行泊松回归模型时出现内存问题。加载数据并为模型做好准备后，python 使用了大约 650 MB 的内存。一旦我创建模型，

import theano.tensor as t
with pm.Model() as poisson_model:
    # priors for coefficients
    coeffs = pm.Uniform('coeffs', -10, 10, shape=(1, predictors.shape[1]))

    r = t.exp(pm.sum(coeffs*predictors.values, 1))

    obs = pm.Poisson('obs', r, observed=targets)

内存使用量高达 3 GB。只有 350 个 8 位整数的数据点，所以我不知道什么在使用这么多的内存。

在玩了一会儿之后，我发现向模型添加任何东西都会使其内存达到 3 GB，甚至像这样简单的东西

with pm.Model() as poisson_model:
    test = pm.Uniform('test', -1, 1)

关于正在发生的事情或我如何看得更深的任何建议？我正在使用新的 iMac、Python 2.7 和最新版本的 PyMC3。谢谢。

4

1 回答 1

1

我尝试在我的系统（Macbook Air，Py 2.7）上复制它，但内存使用量约为 80MB。我会尝试几件事：

  1. 清除 theano 缓存：

    theano缓存清除

  2. 尝试更新 Theano

  3. 从主分支重新安装 PyMC

这些都是猜测，因为我无法复制这个问题，所以我希望其中一个能解决问题。

于 2014-09-17T13:29:18.633 回答