dataeval.core.rank_result_class_balanced

dataeval.core.rank_result_class_balanced(result, class_labels)

Transform RankResult indices using class-balanced selection.

Takes a RankResult and reranks to ensure balanced representation across classes.

Parameters:
result : RankResult

Ranking result (assumed easy_first order).

class_labels : NDArray[np.integer]

Class label for each sample in the original dataset.

Returns:

Reordered indices in easy_first order with class balance applied.

Return type:

NDArray[np.intp]