voiage.methods.heterogeneity.identify_optimal_subgroups

identify_optimal_subgroups

identify_optimal_subgroups([positional or keyword] result: HeterogeneityResult = None) -> dict[str, str]

Return the optimal strategy name for each subgroup.

Parameters

result : HeterogeneityResult Result produced by :func:value_of_heterogeneity.

Returns

dict[str, str] Mapping from subgroup label to subgroup-optimal strategy name.

Examples

>>> import numpy as np >>> from voiage.methods.heterogeneity import identify_optimal_subgroups, value_of_heterogeneity >>> from voiage.schema import ValueArray >>> result = value_of_heterogeneity( … ValueArray.from_numpy(np.array([[10.0, 8.0], [6.0, 12.0]]), [“A”, “B”]), … subgroups=[“low”, “high”], … ) >>> identify_optimal_subgroups(result) {‘high’: ‘B’, ‘low’: ‘A’}

Parameters:

• result HeterogeneityResult

Returns: dict[str, str]