scib_metrics.utils.one_hot

scib_metrics.utils.one_hot(y, n_classes=None)

One-hot encode an array. Wrapper around one_hot().

Parameters:

• y (Union[ndarray, Array]) – Array of shape (n_cells,) or (n_cells, 1).

• n_classes (Optional[int] (default: None)) – Number of classes. If None, inferred from the data.

Return type:

Array

Returns:

one_hot: jnp.ndarray Array of shape (n_cells, n_classes).