View source on GitHub
|
Methods and objectives for variational inference.
Modules
mutual_information module: Mutual information estimators and helpers.
Classes
class GradientEstimators: Gradient estimators for variational losses.
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(log(Avg(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.
fit_surrogate_posterior_stateless(...): 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_variational_loss(...): Monte-Carlo approximation of an f-Divergence variational loss. (deprecated arguments)
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.