Skip to content

Commit 11afbed

Browse files
committed
update apo scripts
1 parent 2f53537 commit 11afbed

File tree

4 files changed

+128
-0
lines changed

4 files changed

+128
-0
lines changed

scripts/irm/apo.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
from montecover.irm import APOCoverageSimulation
2+
3+
# Create and run simulation with config file
4+
sim = APOCoverageSimulation(
5+
config_file="scripts/irm/apo_config.yml",
6+
log_level="INFO",
7+
log_file="logs/irm/apo_sim.log",
8+
)
9+
sim.run_simulation()
10+
sim.save_results(output_path="results/irm/", file_prefix="apo")
11+
12+
# Save config file for reproducibility
13+
sim.save_config("results/irm/apo_config.yml")

scripts/irm/apo_config.yml

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
# Simulation parameters for APO Coverage
2+
3+
simulation_parameters:
4+
repetitions: 1000
5+
max_runtime: 19800 # 5.5 hours in seconds
6+
random_seed: 42
7+
n_jobs: -2
8+
9+
dgp_parameters:
10+
n_obs: [500] # Sample size
11+
n_levels: [2]
12+
linear: [True]
13+
14+
# Define reusable learner configurations
15+
learner_definitions:
16+
linear: &linear
17+
name: "Linear"
18+
19+
logit: &logit
20+
name: "Logistic"
21+
22+
lgbmr: &lgbmr
23+
name: "LGBM Regr."
24+
params:
25+
n_estimators: 500
26+
learning_rate: 0.01
27+
min_child_samples: 10
28+
29+
lgbmc: &lgbmc
30+
name: "LGBM Clas."
31+
params:
32+
n_estimators: 500
33+
learning_rate: 0.01
34+
min_child_samples: 10
35+
36+
dml_parameters:
37+
treatment_level: [0, 1, 2]
38+
trimming_threshold: [0.01]
39+
learners:
40+
- ml_g: *linear
41+
ml_m: *logit
42+
- ml_g: *lgbmr
43+
ml_m: *lgbmc
44+
- ml_g: *lgbmr
45+
ml_m: *logit
46+
- ml_g: *linear
47+
ml_m: *lgbmc
48+
49+
50+
confidence_parameters:
51+
level: [0.95, 0.90] # Confidence levels

scripts/irm/apos.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
from montecover.irm import APOSCoverageSimulation
2+
3+
# Create and run simulation with config file
4+
sim = APOSCoverageSimulation(
5+
config_file="scripts/irm/apos_config.yml",
6+
log_level="INFO",
7+
log_file="logs/irm/apos_sim.log",
8+
)
9+
sim.run_simulation()
10+
sim.save_results(output_path="results/irm/", file_prefix="apos")
11+
12+
# Save config file for reproducibility
13+
sim.save_config("results/irm/apos_config.yml")

scripts/irm/apos_config.yml

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
# Simulation parameters for APOS Coverage
2+
3+
simulation_parameters:
4+
repetitions: 1000
5+
max_runtime: 19800 # 5.5 hours in seconds
6+
random_seed: 42
7+
n_jobs: -2
8+
9+
dgp_parameters:
10+
n_obs: [500] # Sample size
11+
n_levels: [2]
12+
linear: [True]
13+
14+
# Define reusable learner configurations
15+
learner_definitions:
16+
linear: &linear
17+
name: "Linear"
18+
19+
logit: &logit
20+
name: "Logistic"
21+
22+
lgbmr: &lgbmr
23+
name: "LGBM Regr."
24+
params:
25+
n_estimators: 500
26+
learning_rate: 0.01
27+
min_child_samples: 10
28+
29+
lgbmc: &lgbmc
30+
name: "LGBM Clas."
31+
params:
32+
n_estimators: 500
33+
learning_rate: 0.01
34+
min_child_samples: 10
35+
36+
dml_parameters:
37+
treatment_levels: [[0, 1, 2]]
38+
trimming_threshold: [0.01]
39+
learners:
40+
- ml_g: *linear
41+
ml_m: *logit
42+
- ml_g: *lgbmr
43+
ml_m: *lgbmc
44+
- ml_g: *lgbmr
45+
ml_m: *logit
46+
- ml_g: *linear
47+
ml_m: *lgbmc
48+
49+
50+
confidence_parameters:
51+
level: [0.95, 0.90] # Confidence levels

0 commit comments

Comments
 (0)