skills/experimental-design/scripts/randomization.py

• _normalize_ratio · function · L26-L38 — def _normalize_ratio(arms, ratio)
• simple_randomization · function · L41-L54 — def simple_randomization(n, arms=("treatment", "control"), ratio=None, seed=0)
• block_randomization · function · L57-L86 — def block_randomization(n, arms=("treatment", "control"), block_size=None, ratio=None, seed=0)
• stratified_block_randomization · function · L89-L112 — def stratified_block_randomization(strata, arms=("treatment", "control"), block_size=None, ratio=None, seed=0)
• cluster_randomization · function · L115-L132 — def cluster_randomization(clusters, arms=("treatment", "control"), ratio=None, block_size=None, seed=0)
• assign_factorial_runs · function · L135-L147 — def assign_factorial_runs(design_df, seed=0)
• arm_balance · function · L150-L154 — def arm_balance(df, arm_col="arm", by=None)