btb.selection.hierarchical module

class btb.selection.hierarchical.HierarchicalByAlgorithm(choices, by_algorithm)

Bases: btb.selection.ucb1.UCB1

Hierarchical selector

Parameters

by_algorithm (Dict[str, List]) – mapping of ML algorithms to frozen set choices

select(choice_scores)

Groups the frozen sets by algorithm and first chooses an algorithm based on the traditional UCB1 criteria.

Next, from that algorithm’s frozen sets, makes the final set choice.