tfep.tests.nn.check_autoregressive_property

tfep.tests.nn.check_autoregressive_property(model: Module, x: Tensor, degrees_in: Sequence[int], degrees_out: Sequence[int])[source]

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

The test will pass if all the outputs will depend on the inputs such that degree_in < degree_out.

Parameters:

  • model (torch.nn.Module) – The model.

  • x (torch.Tensor) – Shape (n_features_in,). Input.

  • degrees_in (Sequence[int]) – Shape (n_features_in,). Degrees for the input features.

  • degrees_out (Sequence[int]) – Shape (n_features_out,). Degrees for the output features.