Merge branch 'master' of https://github.com/davidavdav/GaussianMixtur…

…es.jl
davidavdav · Mar 1, 2020 · 77b72b5 · 77b72b5
2 parents dd17737 + 604637c
commit 77b72b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/distributions.jl b/src/distributions.jl
@@ -22,11 +22,11 @@ end
 ## conversion to MixtureModel
 function Distributions.MixtureModel(gmm::GMM{T}) where {T<:AbstractFloat}
     if gmm.d == 1
-        mixtures = [Normal(gmm.μ[i,1], gmm.Σ[i,1]) for i=1:gmm.n]
+        mixtures = [Normal(gmm.μ[i,1], sqrt(gmm.Σ[i,1])) for i=1:gmm.n]
     elseif kind(gmm) == :full
         mixtures = [MvNormal(vec(gmm.μ[i,:]), covar(gmm.Σ[i])) for i=1:gmm.n]
     else
-        mixtures = [MvNormal(vec(gmm.μ[i,:]), vec(gmm.Σ[i,:])) for i=1:gmm.n]
+        mixtures = [MvNormal(vec(gmm.μ[i,:]), sqrt(vec(gmm.Σ[i,:]))) for i=1:gmm.n]
     end
     MixtureModel(mixtures, gmm.w)
 end