generators module

Tools to build out-of-the box models

create_checkerboard_hypercube_realnvp(d, repetitions=2, d_hidden=256, n_hidden=16)[source]
create_hypercube_flow(d, model_type='realnvp', mask_type='checkerboard', **params)[source]

Provide a high-level generator with default behavior

create_n_ary_hypercube_flow(cell, d, n=2, repetitions=2, **cell_options)[source]

Create a sequential normalizing flow on the d-dimensional unit hypercube based on a given coupling cell with a n-ary mask. Subsequent cells have masks offset by one at each step until all possible masks have been used.

Parameters

  • cell – coupling cell class

  • d (int) – dimension of the unit hypercube

  • n – period of the n-ary mask

  • repetitions – how many times a full repetition through all possible masks is repeated

  • cell_options – options to be passed at the coupling cell instantiation

Returns

Return type

InvertibleSequentialFlow

create_n_ary_hypercube_realnvp(d, n=2, repetitions=2, d_hidden=256, n_hidden=16)[source]