Commit 2515737e authored by Emmanuel PIGNAT's avatar Emmanuel PIGNAT
Browse files

bug when retrieving GMM output of MTMM conditinoing

parent b9dfd56d
......@@ -237,7 +237,7 @@ class MTMM(GMM):
# the conditional distribution is now a still a mixture
if return_gmm:
return mu_est, sigma_est * nu/(nu-2.)
return h, mu_est, sigma_est * (nu/(nu-2.))[:, None, None, None]
elif return_linear:
As = inv_sigma_out_in
bs = mu_out - np.matmul(inv_sigma_out_in, mu_in[:, :, None])[:, :, 0]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment