diff --git a/src/classic_em.jl b/src/classic_em.jl index 9a02929..ffe5aee 100644 --- a/src/classic_em.jl +++ b/src/classic_em.jl @@ -5,10 +5,11 @@ The EM algorithm was introduced by A. P. Dempster, N. M. Laird and D. B. Rubin i struct ClassicEM <: AbstractEM end """ - fit_mle!(α::AbstractVector, dists::AbstractVector{F} where {F<:Distribution}, y::AbstractVecOrMat, method::ClassicEM; display=:none, maxiter=1000, atol=1e-3, robust=false) + fit_mle!(α::AbstractVector, dists::AbstractVector{F} where {F<:Distribution}, y::AbstractVecOrMat, method::ClassicEM; display=:none, maxiter=1000, atol=1e-3, rtol=nothing, robust=false) Use the EM algorithm to update the Distribution `dists` and weights `α` composing a mixture distribution. - `robust = true` will prevent the (log)likelihood to overflow to `-∞` or `∞`. -- `atol` criteria determining the convergence of the algorithm. If the Loglikelihood difference between two iteration `i` and `i+1` is smaller than `atol` i.e. `|ℓ⁽ⁱ⁺¹⁾ - ℓ⁽ⁱ⁾|