TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfp.vi

View source on GitHub

Methods and objectives for variational inference.

Modules

mutual_information module: Mutual information estimators and helpers.

Functions

amari_alpha(...): The Amari-alpha Csiszar-function in log-space.

arithmetic_geometric(...): The Arithmetic-Geometric Csiszar-function in log-space.

chi_square(...): The chi-Square Csiszar-function in log-space.

csiszar_vimco(...): Use VIMCO to lower the variance of gradient[csiszar_function(Avg(logu))].

csiszar_vimco_helper(...): Helper to csiszar_vimco; computes log_avg_u, log_sooavg_u. (deprecated)

dual_csiszar_function(...): Calculates the dual Csiszar-function in log-space.

fit_surrogate_posterior(...): Fit a surrogate posterior to a target (unnormalized) log density.

jeffreys(...): The Jeffreys Csiszar-function in log-space.

jensen_shannon(...): The Jensen-Shannon Csiszar-function in log-space.

kl_forward(...): The forward Kullback-Leibler Csiszar-function in log-space.

kl_reverse(...): The reverse Kullback-Leibler Csiszar-function in log-space.

log1p_abs(...): The log1p-abs Csiszar-function in log-space.

modified_gan(...): The Modified-GAN Csiszar-function in log-space.

monte_carlo_csiszar_f_divergence(...): Monte-Carlo approximation of the Csiszar f-Divergence. (deprecated)

monte_carlo_variational_loss(...): Monte-Carlo approximation of an f-Divergence variational loss.

pearson(...): The Pearson Csiszar-function in log-space.

squared_hellinger(...): The Squared-Hellinger Csiszar-function in log-space.

symmetrized_csiszar_function(...): Symmetrizes a Csiszar-function in log-space.

t_power(...): The T-Power Csiszar-function in log-space.

total_variation(...): The Total Variation Csiszar-function in log-space.

triangular(...): The Triangular Csiszar-function in log-space.