Binary

class torcheeg.transforms.Binary(threshold: float)

Binarize the label according to a certain threshold. Labels larger than the threshold are set to 1, and labels smaller than the threshold are set to 0.

transform = Binary(threshold=5.0)
transform(y=4.5)['y']
>>> 0

Binary allows simultaneous binarization using the same threshold for multiple labels.

transform = Binary(threshold=5.0)
transform(y=[4.5, 5.5])['y']
>>> [0, 1]

Parameters

threshold (float) – Threshold used during binarization.

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

Parameters

label (int, float, or list) – The input label or list of labels.

Returns

The output label or list of labels after binarization.

Return type

int, float, or list