File tree Expand file tree Collapse file tree 4 files changed +128
-0
lines changed Expand file tree Collapse file tree 4 files changed +128
-0
lines changed Original file line number Diff line number Diff line change
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" )
Original file line number Diff line number Diff line change
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
Original file line number Diff line number Diff line change
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" )
Original file line number Diff line number Diff line change
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
You can’t perform that action at this time.
0 commit comments