CMD + K

pomegranate

Community

A PyTorch implementation of probabilistic models.

Installation

To install this package, run one of the following:

Conda
$conda install anaconda::pomegranate

Usage Tracking

1.1.2
0.14.4
0.14.3
0.14.2
0.14.0
5 / 8 versions selected
Downloads (Last 6 months): 0

Description

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.

About

Summary

A PyTorch implementation of probabilistic models.

Last Updated

Aug 12, 2025 at 11:00

License

MIT

Supported Platforms

linux-64
linux-aarch64
macOS-64
win-64
macOS-arm64

Unsupported Platforms

linux-ppc64le Last supported version: 0.14.4
win-32 Last supported version: 0.14.4
linux-32 Last supported version: 0.10.0