selfeeg.models

This module collects various Deep Learning models and custom layers. It is divided in two submodules:

• layers: a collection custom layers with the possibility to add norm constraints.

• zoo: a collection of deep learning models proposed for EEG applications.

models.layers module

Classes

ConstrainedConv1d	nn.Conv1d layer with norm constraints.
ConstrainedConv2d	nn.conv2d layer with norm constraints.
ConstrainedDense	nn.Linear layer with norm constraints.
DepthwiseConv2d	Depthwise 2D layer with norm constraints.
FilterBank	Filter bank layer of the FBCNet model.
SeparableConv2d	Separable Convolutional layer with norm constraints.

models.encoders module

Classes

DeepConvNetEncoder	Pytorch Implementation of the DeepConvNet Encoder.
EEGConformerEncoder	Pytorch implementation of the EEGConformer Encoder.
EEGInceptionEncoder	Pytorch Implementation of the EEGInception Encoder.
EEGNetEncoder	Pytorch Implementation of the EEGnet Encoder.
EEGSymEncoder	Pytorch implementation of the EEGSym Encoder.
FBCNetEncoder	Pytorch implementation of the FBCNet Encoder.
ResNet1DEncoder	Pytorch implementation of the Resnet Encoder
ShallowNetEncoder	Pytorch implementation of the ShallowNet Encoder.
StagerNetEncoder	Pytorch implementation of the StagerNet Encoder.
STNetEncoder	Pytorch implementation of the STNet Encoder.
TinySleepNetEncoder	Pytorch Implementation of the TinySleepNet Encoder.
xEEGNetEncoder	Pytorch implementation of the xEEGNet Encoder.

models.zoo module

Classes

ATCNet	Pytorch implementation of the ATCNet model.
DeepConvNet	Pytorch Implementation of the DeepConvNet model.
EEGConformer	Pytorch implementation of EEGConformer.
EEGInception	Pytorch Implementation of the EEGInception model.
EEGNet	Pytorch implementation of the EEGNet model.
EEGSym	Pytorch implementation of the EEGSym model.
FBCNet	Pytorch implementation of the FBCNet model.
ResNet1D	Pytorch implementation of the Resnet model
ShallowNet	Pytorch implementation of the ShallowNet model.
StagerNet	Pytorch implementation of the StagerNet model.
STNet	Pytorch implementation of the STNet model.
TinySleepNet	Pytorch Implementation of the TinySleepNet model.
xEEGNet	Pytorch implementation of xEEGNet.