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.

Deepsets

Deepsets Model from this paper .

DeepsetsConfig

Dataclass config for a Deepsets.

Gru

A multi-layer Gated Recurrent Unit (GRU) RNN like the one from torch .

GruConfig

Dataclass config for a Gru.

Lstm

A multi-layer Long Short-Term Memory (LSTM) RNN like the one from torch .

LstmConfig

Dataclass config for a LSTM.