shapiq.utils.get_explicit_subsets

shapiq.utils.get_explicit_subsets(n, subset_sizes)

Enumerates all subsets of the given sizes and returns a one-hot matrix.

Parameters:

• n (int) – number of players.

• subset_sizes (list[int]) – list of subset sizes.

Return type:

ndarray

Returns:

one-hot matrix of all subsets of certain sizes.

Examples

>>> get_explicit_subsets(n=4, subset_sizes=[1, 2]).astype(int)
array([[1, 0, 0, 0],
       [0, 1, 0, 0],
       [0, 0, 1, 0],
       [0, 0, 0, 1],
       [1, 1, 0, 0],
       [1, 0, 1, 0],
       [1, 0, 0, 1],
       [0, 1, 1, 0],
       [0, 1, 0, 1],
       [0, 0, 1, 1]])