#125125th of 206 in ML frameworks and libraries

pgmpy

Python library for causal discovery and inference with Bayesian networks

Stars
3.4k
License
MIT
Last commit
Nov 2026
Last release
Apr 2026
Language
Python

Missed the latest GitHub check; these facts may be older.

Overview

pgmpy is a Python library with data structures for DAGs, PDAGs, MAGs, PAGs, Bayesian Networks, Dynamic Bayesian Networks and Structural Equation Models. It covers structure learning (for example the PC estimator), parameter estimation, probabilistic and causal inference, model validation and data simulation. Estimators follow a composable API that is scikit-learn compatible where possible.

Who it is for: Data scientists and researchers doing causal or probabilistic modeling

Strengths

  • Covers discrete, linear Gaussian and functional (mixed) Bayesian networks in one API
  • Includes structure learning, parameter fitting, inference and simulation in the same library
  • Ships example datasets and models, such as bnlearn alarm and ecoli70
  • Installs from PyPI or conda-forge; MIT license

Weaknesses

  • Functional Bayesian Networks need the torch backend and pyro
  • Not an LLM tool; no agent, model-serving or LLM integration
  • Learning and inference examples for functional networks live in a separate notebook, not the README

What it needs

  • Needs torch and pyro (Functional Bayesian Networks only)
  • Models: DAG, PDAG, MAG, PAG, Discrete Bayesian Network

Also in ML frameworks and libraries

See all 206
Also in ML frameworks and libraries
RankProjectScore
1transformersPython library defining open models for text, vision, audio and multimodal tasks168.9k stars+1,573 (+0.9%) in 30 daysApache-2.094 out of 100
2ultralyticsPython library and CLI for training and running YOLO vision models63.3k stars+1,004 (+1.6%) in 30 daysAGPL-3.094 out of 100
3pytorchPython tensor library with GPU support and tape-based autograd105.5k stars+1,036 (+1.0%) in 30 dayscustom license93 out of 100
#4pytorch-image-modelsPyTorch library of image models, pretrained weights and training scripts37.3k stars+107 (+0.3%) in 30 daysApache-2.093 out of 100
#5supervisionPython toolkit for loading, annotating and counting computer vision detections Live demo ↗ (opens in a new tab)51.6k stars+521 (+1.0%) in 30 daysMIT92 out of 100