oryx.core.ppl.intervene

View source on GitHub

Transforms a program into one where provided random variables are fixed.

oryx.core.ppl.intervene(
    f: oryx.core.ppl.LogProbFunction,
    **observations
) -> oryx.core.ppl.LogProbFunction

Used in the notebooks

intervene is a probabilistic program transformation that fixes the values for certain random samples in an input program. A probabilistic program may sample intermediate latent random variables while computing its output. Observing those random variables converts them into deterministic constants that are just used in the forward computation.

Random variables that are intervened are no longer random variables. This means that if a variable x is intervened , it will no longer appear in the joint_sample of a program and its log_prob will no longer be computed as part of a program's log_prob.

Examples:

Simple usage:

def model(key):
  return random_variable(random.normal, name='x')(key)
intervene(model, x=1.)(random.PRNGKey(0))  # => 1.

Multiple random variables:

def model(key):
  k1, k2 = random.split(key)
  z = random_variable(random.normal, name='z')(k1)
  return z + random_variable(random.normal, name='x')(k2)
intervene(model, z=1., x=1.)(random.PRNGKey(0))  # => 2.