Skip to content

Conversation

@el19yz
Copy link

@el19yz el19yz commented Dec 10, 2021

Adding in GaussianMixture distribution with E-M fitting functionality, tests, and notebook demonstrations.

Maximum likelihood estimation of MVN distribution

  • Unweighted ML estimation of MVN full-covariance distribution
  • Weighted ML estimation of MVN full-covariance distribution
  • Unweighted and weighted ML estimation of MVN diagonal cov distribution
  • Unweighted and weighted ML estimation of MVN spherical cov distribution

Expectation maximization inference

  • Create barebones GaussianMixture distribution and sample
  • Test EM inference with full-covariance MVN distribution
  • + diagonal-covariance MVN components distribution
  • + spherical-covariance MVN components distribution

Sidequests

  • Compare home-brewed k-means implementation (see gimbal-refactor repo) with sklearn.cluster.KMeans (additional benefits: does not require reshaping dataset. perhaps can generalize to different distances, i.e. cosine distance to permit generalization to spherical k-means)
  • Propose simpler one_hot implementation

@schlagercollin
Copy link
Member

Nice!

@el19yz el19yz self-assigned this Dec 10, 2021
Verified and tested for MultivariateNormalFullCovariance distribution
@el19yz el19yz requested a review from slinderman December 14, 2021 17:43
@lindermanlab lindermanlab deleted a comment from github-actions bot Dec 14, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants