benchmarl.models
Here you can find the model table.
Common
Abstract class representing a model. |
|
Dataclass representing a |
|
A sequence of |
|
Dataclass for a |
Models
Multi layer perceptron model. |
|
Dataclass config for a |
|
A GNN model. |
|
Dataclass config for a |
|
Convolutional Neural Network (CNN) model. |
|
Dataclass config for a |
|
Deepsets Model from this paper . |
|
Dataclass config for a |
|
A multi-layer Gated Recurrent Unit (GRU) RNN like the one from torch . |
|
Dataclass config for a |
|
A multi-layer Long Short-Term Memory (LSTM) RNN like the one from torch . |
|
Dataclass config for a |