mimo_inception_time#

torch_uncertainty.models.mimo_inception_time(in_channels, num_classes, num_estimators, kernel_size=40, embed_dim=32, num_blocks=6, dropout=0.0, residual=True)[source]#

MIMO InceptionTime.

Parameters:

  • in_channels (int) – Number of input channels.

  • num_classes (int) – Number of output classes.

  • num_estimators (int) – Number of estimators for MIMO.

  • kernel_size (int) – Size of the convolutional kernel. Defaults to 40.

  • embed_dim (int) – Dimension of the embedding. Defaults to 32.

  • num_blocks (int) – Number of inception blocks. Defaults to 6.

  • dropout (float) – Dropout rate. Defaults to 0.0.

  • residual (bool) – Whether to use residual connections. Defaults to True.

Returns:

The MIMO InceptionTime model.

Return type:

_MIMOInceptionTime