广义加法模型 Python 库

机器算法验证 广义加法模型
2022-01-28 20:18:21

我知道 R 有用于广义加法模型的 gam 和 mgcv 库。但是我很难在 Python 生态系统中找到它们的对应物(statsmodels 在沙箱中只有原型)。有人知道现有的 python 库吗?如果不是,谁知道这可能是一个开发/贡献于 scikit-learn 的好项目。

3个回答

我使用惩罚 B 样条编写了 GAM 的 Python 实现。

在这里查看:https ://github.com/dswah/pyGAM

我已经包含了很多链接功能、分布和特性。

另一个快速试验 GAM 模型的选项是包https://github.com/malmgrek/gammy

重点是 GAM 系数的贝叶斯建模以及自定义基函数的易扩展性。目前,例如高斯过程、B 样条以及不同的平凡构造。

最近的另一个发展是将 GAM 方法应用于深度学习架构的神经加法模型:

https://arxiv.org/abs/2004.13912

https://github.com/nickfrosst/neural_additive_models