sleepless.models.ninel_1d_cnn_ssc#

Classes

NineL_1DCNN_SSC(n_channels, n_hidden_state, ...)

Sleep staging architecture from [Satapathy-2023].
class sleepless.models.ninel_1d_cnn_ssc.NineL_1DCNN_SSC(n_channels, n_hidden_state, n_classes)[source]#

Bases: Module

Sleep staging architecture from [Satapathy-2023]. Convolutional neural network for sleep staging described in [Satapathy-2023].

Parameters:

  • n_channels (int) – Number of EEG channels.

  • n_classes (int) – Number of classes.

  • n_hidden_state (int) – output size of the feature extractor

References

forward(x)[source]#

Forward pass.

Parameters:

x – Batch of signals windows of shape (batch_size, n_channels, n_times).