|View source on GitHub|
Takes Edward probabilistic program and returns its log joint function.
tfp.edward2.make_log_joint_fn( model )
Python callable which executes the generative process of a
computable probability distribution using
A log-joint probability function. Its inputs are
Below we define Bayesian logistic regression as an Edward program,
representing the model's generative process. We apply
order to represent the model in terms of its joint probability function.
from tensorflow_probability import edward2 as ed def logistic_regression(features): coeffs = ed.Normal(loc=0., scale=1., sample_shape=features.shape, name="coeffs") outcomes = ed.Bernoulli(logits=tf.tensordot(features, coeffs, [, ]), name="outcomes") return outcomes log_joint = ed.make_log_joint_fn(logistic_regression) features = tf.random.normal([3, 2]) coeffs_value = tf.random.normal() outcomes_value = tf.round(tf.random.uniform()) output = log_joint(features, coeffs=coeffs_value, outcomes=outcomes_value)