FixCategory

class torcheeg.transforms.FixCategory(value: Union[int, str, List])

Returns a pre-set label for all samples, usually used to supplement the dataset with new categories.

transform = FixCategory(value=0)
transform(y=3)['y']
>>> 0

FixCategory allows multiple values to be selected and returned as a list. Suitable for multi-classification tasks or multi-task learning.

transform = FixCategory(value=[0, 1])
transform(y=[1, 2])['y']
>>> [0, 1]

Parameters

value (str or list) – The pre-set label.

__call__(*args, y: Dict, **kwargs) → Union[int, float, List]

Parameters

y (any) – A label or label list describing the EEG signal samples.

Returns

FixCategoryeded value pre-set by value.

Return type

any