zuko.flows.polynomial#

Polynomial flows.

Classes#

SOSPF

Creates a sum-of-squares polynomial flow (SOSPF).

Descriptions#

class zuko.flows.polynomial.SOSPF(features, context=0, degree=3, polynomials=2, **kwargs)#

Creates a sum-of-squares polynomial flow (SOSPF).

Warning

Invertibility is only guaranteed for features within the interval \([-10, 10]\). It is recommended to standardize features (zero mean, unit variance) before training.

See also

zuko.transforms.SOSPolynomialTransform

References

Sum-of-Squares Polynomial Flow (Jaini et al., 2019)
https://arxiv.org/abs/1905.02325
Parameters:

  • features (int) – The number of features.

  • context (int) – The number of context features.

  • degree (int) – The degree \(L\) of polynomials.

  • polynomials (int) – The number of polynomials \(K\).

  • kwargs – Keyword arguments passed to zuko.flows.autoregressive.MAF.