generate_ranks

generate_ranks(num_candidates, prefix_shape=(), seed=None, dtype=None)

Generate random ranks from a given array of the number of candidates for each ranking task.

Parameters:

• num_candidates (ndarray) – shape: s the number of candidates

• prefix_shape (Tuple[int, …]) – additional dimensions for broadcasted sampling

• seed (Union[None, int, Generator]) – the random seed

• dtype (Optional[Type[number]]) – the data type

Return type:

ndarray

Returns:

shape: dims + s an array of sampled rank values