API Reference

Generative Classification

class vbll.layers.classification.GenClassification(*args: Any, **kwargs: Any)

Discriminative Classification

class vbll.layers.classification.DiscClassification(*args: Any, **kwargs: Any)

Regression

class vbll.layers.regression.Regression(*args: Any, **kwargs: Any)

Variational Bayesian Linear Regression

Parameters

in_featuresint

Number of input features

out_featuresint

Number of output features

regularization_weightfloat

Weight on regularization term in ELBO

parameterizationstr

Parameterization of covariance matrix. Currently supports ‘dense’ and ‘diagonal’

prior_scalefloat

Scale of prior covariance matrix

wishart_scalefloat

Scale of Wishart prior on noise covariance

doffloat

Degrees of freedom of Wishart prior on noise covariance