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 ndarray | Array

Array of shape (n_cells,) or (n_cells, 1).

n_classes int | None (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).