Sum distributions based on another distribution

Questions
2

You can use pm.math.switch() to build conditional logic. So maybe something along these lines?

d = pm.Categorical('d', [0.1, 0.85, 0.05])
b_logic = pm.math.switch(pm.math.eq(d, 1), 0, b)
c_logic = pm.math.switch(pm.math.eq(d, 2), 0, c)
sum = pm.Deterministic('sum', a + b_logic + c_logic)

Not pretty, but hopefully it helps.

3 Likes