@@ -44,13 +44,21 @@ function condition(gmm::GMM, conditioned_values::NamedTuple)
44
44
return ConditionedGMM (gmm. data, conditioned_values)
45
45
end
46
46
47
- function LogDensityProblems. logdensity (gmm:: ConditionedGMM{names} , params:: AbstractVector ) where {names}
47
+ function LogDensityProblems. logdensity (
48
+ gmm:: ConditionedGMM{names} , params:: AbstractVector
49
+ ) where {names}
48
50
if Set (names) == Set ([:μ , :w ]) # conditioned on μ, w, so params are z
49
- return log_joint (; μ= gmm. conditioned_values. μ, w= gmm. conditioned_values. w, z= params, x= gmm. data. x)
51
+ return log_joint (;
52
+ μ= gmm. conditioned_values. μ, w= gmm. conditioned_values. w, z= params, x= gmm. data. x
53
+ )
50
54
elseif Set (names) == Set ([:z , :w ]) # conditioned on z, w, so params are μ
51
- return log_joint (; μ= params, w= gmm. conditioned_values. w, z= gmm. conditioned_values. z, x= gmm. data. x)
55
+ return log_joint (;
56
+ μ= params, w= gmm. conditioned_values. w, z= gmm. conditioned_values. z, x= gmm. data. x
57
+ )
52
58
elseif Set (names) == Set ([:z , :μ ]) # conditioned on z, μ, so params are w
53
- return log_joint (; μ= gmm. conditioned_values. μ, w= params, z= gmm. conditioned_values. z, x= gmm. data. x)
59
+ return log_joint (;
60
+ μ= gmm. conditioned_values. μ, w= params, z= gmm. conditioned_values. z, x= gmm. data. x
61
+ )
54
62
else
55
63
error (" Unsupported conditioning configuration." )
56
64
end
0 commit comments