English

pgmpy: A Python Toolkit for Bayesian Networks

Machine Learning 2023-04-19 v1 Methodology

Abstract

Bayesian Networks (BNs) are used in various fields for modeling, prediction, and decision making. pgmpy is a python package that provides a collection of algorithms and tools to work with BNs and related models. It implements algorithms for structure learning, parameter estimation, approximate and exact inference, causal inference, and simulations. These implementations focus on modularity and easy extensibility to allow users to quickly modify/add to existing algorithms, or to implement new algorithms for different use cases. pgmpy is released under the MIT License; the source code is available at: https://github.com/pgmpy/pgmpy, and the documentation at: https://pgmpy.org.

Keywords

Cite

@article{arxiv.2304.08639,
  title  = {pgmpy: A Python Toolkit for Bayesian Networks},
  author = {Ankur Ankan and Johannes Textor},
  journal= {arXiv preprint arXiv:2304.08639},
  year   = {2023}
}