cascade.model.CASCADE.design_brute_force
- CASCADE.design_brute_force(source, target, pool=None, design_size=1, k=30, counterfactual_kws=None, neighbor_kws=None)[source]
Intervention design with brute-force exhaustion
- Parameters:
source (
AnnData) – Source datasettarget (
AnnData) – Target dataset representing desired outcomepool (
list[str] |None) – Optional list of variables as candidate pooldesign_size (
int) – Maximal combinatorial order to considerk (
int) – Number of samples to generate for each designcounterfactual_kws (
Mapping[str,Any] |None) – Additional keyword arguments passed tocounterfactual()neighbor_kws (
Mapping[str,Any] |None) – Additional keyword arguments passed toNearestNeighbors
- Return type:
- Returns:
DataFrame of intervention designs, sorted by descending vote counts
AnnData object with counterfactual predictions for all designs