Skip to content

Commit fa9822f

Browse files
author
Alexander Ororbia
committed
minor edit to gmm/bmm docs
1 parent 9e327e1 commit fa9822f

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

ngclearn/utils/density/bmm.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,9 @@ class BMM: ## Bernoulli mixture model (mixture-of-Bernoullis)
6262
"""
6363
Implements a Bernoulli mixture model (BMM) -- or mixture of Bernoullis (MoB).
6464
Adaptation of parameters is conducted via the Expectation-Maximization (EM)
65-
learning algorithm and leverages full covariance matrices in the component
66-
multivariate Bernoulli distributions.
65+
learning algorithm. Note that this Bernoulli mixture assumes that each component
66+
is a factorizable mutlivariate Bernoulli distribution. (A Categorical distribution
67+
is assumed over the latent variables).
6768
6869
Args:
6970
K: the number of components/latent variables within this BMM

ngclearn/utils/density/gmm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,8 @@ class GMM: ## Gaussian mixture model (mixture-of-Gaussians)
104104
Implements a Gaussian mixture model (GMM) -- or mixture of Gaussians (MoG).
105105
Adaptation of parameters is conducted via the Expectation-Maximization (EM)
106106
learning algorithm and leverages full covariance matrices in the component
107-
multivariate Gaussians.
107+
multivariate Gaussians. (A Categorical distribution is assumed over the
108+
latent variables).
108109
109110
Args:
110111
K: the number of components/latent variables within this GMM

0 commit comments

Comments
 (0)