benchmarl.models.DeepsetsConfig

class DeepsetsConfig(aggr: str, out_features_local_nn: int, local_nn_num_cells: Sequence[int], local_nn_activation_class: Type[Module], global_nn_num_cells: Sequence[int], global_nn_activation_class: Type[Module])[source]

Bases: ModelConfig

Dataclass config for a Deepsets.

static associated_class()[source]

The associated Model class