setu.one_hot

setu.one_hot(ids, num_classes)

One-hot encode integer IDs.

Parameters:

• ids (Array) – Integer array of class IDs.

• num_classes (int) – Total number of classes.

Return type:

Array

Returns:

One-hot encoded array with shape (*ids.shape, num_classes).