Function - one_hot

Contents

Function - one_hot#

openfl.utilities.data_splitters.numpy.one_hot(labels, classes)[source]#

Apply One-Hot encoding to labels.

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

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

Returns:

The one-hot encoded labels.

Return type:

np.ndarray