cascade.nn.Func

class cascade.nn.Func(in_features, cov_features, out_features, hidden_dim, n_layers, multi_dims, dropout)

Bases: Module

Structural equation with covariates

Parameters:

• in_features (int) – Input dimensionality

• cov_features (int) – Covariate dimensionality

• out_features (int) – Output dimensionality

• hidden_dim (int) – Hidden layer dimensionality

• n_layers (int) – Number of hidden layers

• multi_dims (tuple[int, ...]) – Multiplex dims at the front of input samples

• dropout (float) – Dropout rate

Methods

forward