diff --git a/Project.toml b/Project.toml index e8c494d..fdb9d66 100644 --- a/Project.toml +++ b/Project.toml @@ -10,6 +10,7 @@ StatsFuns = "4c63d2b9-4356-54db-8cca-17b64c39e42c" [compat] Distributions = "0.25" +ArgCheck = "2" julia = "1" [extras] diff --git a/README.md b/README.md index fbf8d2b..aa45523 100644 --- a/README.md +++ b/README.md @@ -3,6 +3,8 @@ This package provides a simple implementation of the Expectation Maximization algorithm used to fit mixture models. Due to [Julia](https://julialang.org/) amazing [multiple dispatch](https://www.youtube.com/watch?v=kc9HwsxE1OY) systems and the [Distributions](https://juliastats.org/Distributions.jl/stable/) package, the code is very generic i.e., mixture of all common distributions should be supported. +I plan to add different methods for E-step and M-steps like stochastic EM and others. + ## Example ```julia