peal.visualization.model_comparison.create_checkbox_dict

peal.visualization.model_comparison.create_checkbox_dict(criterion_keys)

Enumerate all combinations of binary criteria as checkbox columns.

Parameters:

criterion_keys (iterable of str) – Names of the criteria (n of them).

Returns:

{criterion: torch.LongTensor of shape (2 ** n,)} where entry i says whether the criterion holds in combination i (the flattened [2] * n index grid).

Return type:

dict