pomegranate
A PyTorch implementation of probabilistic models.
A PyTorch implementation of probabilistic models.
To install this package, run one of the following:
pomegranate is a library for probabilistic modeling defined by its modular implementation and treatment of all models as the probability distributions they are. The modular implementation allows one to easily drop normal distributions into a mixture model to create a Gaussian mixture model just as easily as dropping a gamma and a Poisson distribution into a mixture model to create a heterogeneous mixture. But that's not all! Because each model is treated as a probability distribution, Bayesian networks can be dropped into a mixture just as easily as a normal distribution, and hidden Markov models can be dropped into Bayes classifiers to make a classifier over sequences. Together, these two design choices enable a flexibility not seen in any other probabilistic modeling package.
Summary
A PyTorch implementation of probabilistic models.
Last Updated
Aug 12, 2025 at 11:00
License
MIT
Supported Platforms
Unsupported Platforms
GitHub Repository
https://github.com/jmschrei/pomegranateDocumentation
https://pomegranate.readthedocs.io