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: int | float | List, **kwargs) → 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