tfep.tests.nn

Test objects and function of the tfep.nn package.

Functions

check_autoregressive_property(model, x, ...)

Raises an AssertionError if y = model(x) does not satisfy the autoregressive property.

create_random_input(batch_size, n_features)

Create random input and parameters.

Modules

conditioners

embeddings

flows

transformers