Skip to content

MixtureDensityNetworks v0.2.2

Choose a tag to compare

View all tags
@JoshuaBillson JoshuaBillson released this 29 Apr 01:33
· 2 commits to main since this release
v0.2.2
24cc037

This release replaces the full covariance matrix in the MultivariateGMM layer with a diagonal covariance matrix.

Features

  • This version replaces the full covariance matrix in the MultivariateGMM layer with a diagonal covariance matrix. This provides significantly faster training time at the cost of representation power. However, we can regain much of this power by increasing the number of mixture components in the predicted distribution.
Assets 2
Loading