benchmarl.models

Here you can find the model table.

Common

Model

Abstract class representing a model.

ModelConfig

Dataclass representing a Model configuration.

SequenceModel

A sequence of Model

SequenceModelConfig

Dataclass for a SequenceModel.

Models

Mlp

Multi layer perceptron model.

MlpConfig

Dataclass config for a Mlp.

Gnn

A GNN model.

GnnConfig

Dataclass config for a Gnn.

Cnn

Convolutional Neural Network (CNN) model.

CnnConfig

Dataclass config for a Cnn.