3D visualization of GMM learning via the EM algorithm
This is a 3D visualization of how the Expectation Maximization algorithm learns a Gaussian Mixture Model for 3-dimensional data.
--How it works--
The data is either read in or generated in general-covariance gaussian clusters. For desired values of k (number of Gaussians to fit), a movie is played showing the evolution of the GMM through the iterations of the EM algorithm. The true model is only available at each iteration (viewed as an anchor frame in a movie), so the illusion of movement of the 3D Gaussians (displayed as ellipsoids at 1 standard deviation) is given by interpolating enough "frames" in between these anchor frames.
Cite As
Johannes (2024). 3D visualization of GMM learning via the EM algorithm (https://www.mathworks.com/matlabcentral/fileexchange/34527-3d-visualization-of-gmm-learning-via-the-em-algorithm), MATLAB Central File Exchange. Retrieved .
MATLAB Release Compatibility
Platform Compatibility
Windows macOS LinuxCategories
- AI, Data Science, and Statistics > Statistics and Machine Learning Toolbox > Probability Distributions > Multivariate Distributions > Gaussian Mixture Distribution >
Tags
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!Discover Live Editor
Create scripts with code, output, and formatted text in a single executable document.
Version | Published | Release Notes | |
---|---|---|---|
1.9.0.0 | Simplified 2D plotting option, stream-lined code |
||
1.8.0.0 | Cleaned up code, improved plotting, coloring, added full-covariance capabilities |
||
1.6.0.0 | Corrected bug in EM implementation |
||
1.5.0.0 | changed mean initialization from 'at data mean with small noise' to 'at data points chosen at random' (usually gives better clustering) |
||
1.4.0.0 | reduced overhead |
||
1.2.0.0 | updated description |
||
1.1.0.0 | color gradient added to data points to show the certainty that it belongs to the nearest Gaussian |
||
1.0.0.0 |