Expectation Maximization (EM) algo demo in Pytorch

Hi guys,

Here are some demos of how to implement the EM algorithm in Pytorch.

If anyone has also done this, would be super interested to see different takes on it.

1D example

1 Like