View source on GitHub
|
Layers for combining tfp.distributions and tf.keras.
Classes
class CategoricalMixtureOfOneHotCategorical: A OneHotCategorical mixture Keras layer from k * (1 + d) params.
class DistributionLambda: Keras layer enabling plumbing TFP distributions through Keras models.
class IndependentBernoulli: An Independent-Bernoulli Keras layer from prod(event_shape) params.
class IndependentLogistic: An independent logistic Keras layer.
class IndependentNormal: An independent normal Keras layer.
class IndependentPoisson: An independent Poisson Keras layer.
class KLDivergenceAddLoss: Pass-through layer that adds a KL divergence penalty to the model loss.
class KLDivergenceRegularizer: Regularizer that adds a KL divergence penalty to the model loss.
class MixtureLogistic: A mixture distribution Keras layer, with independent logistic components.
class MixtureNormal: A mixture distribution Keras layer, with independent normal components.
class MixtureSameFamily: A mixture (same-family) Keras layer.
class MultivariateNormalTriL: A d-variate MVNTriL Keras layer from d + d * (d + 1) // 2 params.
class OneHotCategorical: A d-variate OneHotCategorical Keras layer from d params.
class VariationalGaussianProcess: A VariationalGaussianProcess Layer.