pgmpy/pgmpy
Python Toolkit for Causal and Probabilistic Reasoning
解决的问题
pgmpy 是一个使用图形模型进行因果和概率推理的库。它允许用户构建、学习并执行对复杂变量间依赖关系进行建模的推理,从而实现因果关系的发现以及基于观测证据的概率计算。
工作原理
该库实现了多种图形模型的数据结构,包括贝叶斯网络、动态贝叶斯网络和结构方程模型。它提供了一个模块化、可组合的 API,用于执行以下任务:
- 因果发现:从数据或专家知识中学习模型结构。
- 参数学习:从观测数据中估计条件概率分布。
- 概率推理:基于证据计算后验分布。
- 因果推断:计算干预分布。
- 模拟:在特定干预或证据下生成合成数据。
适用人群
专为需要在 Python 中进行概率图建模、因果发现和因果推断的研究人员和开发者设计,兼容 scikit-learn 的流水线。
主要亮点
- 多样化的模型支持:支持 DAG、PDAG、MAG、PAG 以及线性高斯贝叶斯网络。
- 因果工具集成:提供因果发现、识别和验证的集成工具。
- 因果与概率推理:能够计算后验分布和干预分布。
- 与 scikit-learn 兼容:模块化 API 可轻松集成到标准机器学习流水线中。
- 灵活的数据处理:支持离散、线性高斯和混合数据,通过函数型贝叶斯网络可处理任意关系。
相关
- 项目
- 项目
- 项目
- 项目
- 项目