Xception#

class torch_ecg.models.Xception(in_channels: int, **config)[source]#

Bases: Sequential, SizeMixin, CitationMixin

Xception model.

Xception is an architecture that uses depthwise separable convolutions to build light-weight deep neural networks, as described in [1]. Its official implementation is available in [2], and a PyTorch implementation is available in [3]. Xception is currently not widely used in the field of ECG analysis, but has the potential to be highly effective for this task.

Parameters:

  • in_channels (int) – Number of channels in the input.

  • config (dict) – Other hyper-parameters of the Module, ref. corr. config file. For keyword arguments that must be set in 3 sub-dict, namely in “entry_flow”, “middle_flow”, and “exit_flow”, refer to corr. docstring of each class.

References

compute_features_output_shape(seq_len: int | None = None, batch_size: int | None = None) → Sequence[int | None][source]#

Compute the output shape of the features.

Parameters:

  • seq_len (int, optional) – Length of the input signal tensor.

  • batch_size (int, optional) – Batch size of the input signal tensor.

Returns:

output_shape – Output shape of the features.

Return type:

sequence

compute_output_shape(seq_len: int | None = None, batch_size: int | None = None) → Sequence[int | None][source]#

Compute the output shape the model.

Parameters:

  • seq_len (int, optional) – Length of the input tensors.

  • batch_size (int, optional) – Batch size of the input tensors.

Returns:

output_shape – The output shape of the module.

Return type:

sequence

forward(input: Tensor) → Tensor[source]#

Forward pass of the model.

Parameters:

input (torch.Tensor.) – Input signal tensor, of shape (batch_size, n_channels, seq_len).

Returns:

output – Output tensor, of shape (batch_size, n_channels, seq_len).

Return type:

torch.Tensor

forward_features(input: Tensor) → Tensor[source]#

Forward pass of the model to extract features.

Parameters:

input (torch.Tensor) – Input signal tensor, of shape (batch_size, channels, seq_len).

Returns:

features – Feature map tensor, of shape (batch_size, channels, seq_len).

Return type:

torch.Tensor