openfl.utilities.data_splitters.numpy.one_hot

openfl.utilities.data_splitters.numpy.one_hot(labels, classes)

Apply One-Hot encoding to labels.

Parameters:

• labels (np.ndarray) – Array of labels.

• classes (int) – The total number of classes.

Returns:

np.ndarray – The one-hot encoded labels.