torcheeg.models.rst

torcheeg.models

Extensive baseline method reproduction.

torcheeg.models

Convolutional Neural Networks

EEGNet FBCCNN FBCNet MTCNN STNet TSCeption CCNN SSTEmotionNet

Recurrent Neural Networks

GRU LSTM

Graph Neural Networks

DGCNN LGGNet pyg.RGNN pyg.GIN

Transformer

SimpleViT ArjunViT VanillaTransformer ViT

Generative Adversarial Network

BGenerator BDiscriminator BCGenerator BCDiscriminator

Variational Auto Encoder

BEncoder BDecoder BCEncoder BCDecoder

Normalization Flow

BGlow

Diffusion Models

BUNet BCUNet