pymc3:使用坚果

pymc3: using NUTS

我想在不同的 mcmc 方法之间进行公平比较,特别是,我想比较我的方法与 NUTS 的采样性能。但是,我认为 pymc3 中的采样函数有很多 settings/tuning 提高了采样算法的性能,超越了原来的方法。

所以,我想知道是否有人可以告诉我如何禁用这些调整。简而言之,我只想有一个简单的 NUTS 算法(通过 NUTS,我的意思是算法 6:具有双重平均的高效 No-U-Turn 采样器)

谢谢

这可能很困难——PyMC3 和 Stan(其中一些维护者编写了 NUTS 论文)都结合了新的最佳实践和改进。您可以从 github 克隆它并查看早期实现。

This commit has 一个非常接近论文中的符号的 NUTS 实现。它已被重构以使用更具描述性的变量名称以及上述改进。