Squeeze#

class dpeeg.transforms.Squeeze(key: str = 'edata', dim: int = 1)[source]#

Remove a dimension on the data.

Parameters:

key (str) – The key of the eegdata value to be transformed.
dim (int) – Selects a subset of the entries of length one in the shape. If a dim is selected with shape entry greater than one, an error is raised.

Returns:

data – Transformed eegdata.

Return type:

eegdata or dataset

Examples

>>> eegdata = dpeeg.EEGData(edata=np.random.randn(16, 1, 3, 10),
...                         label=np.random.randint(0, 3, 16))
>>> transforms.Squeeze()(eegdata, verbose=False)
[edata=(16, 3, 10), label=(16,)]