Code for all experiments in the manuscript "Effect-Invariant Mechanisms for Policy Generalization"
Section 6.1: section6-1_simulation.py
Section 6.3: finding_invariant_sets.ipynb
Section 6.4: heartstep_zero_shot_simu.ipynb
For the experiments in Section 6.4, we use the publicly available dataset HeartStepsV1.