add treatment level to apo simulation

SvenKlaassen · SvenKlaassen · commit a68492346e5a · 2025-06-04T14:05:43.000+02:00
diff --git a/monte-cover/src/montecover/irm/apo.py b/monte-cover/src/montecover/irm/apo.py
@@ -107,6 +107,7 @@ def run_single_rep(self, dml_data: dml.DoubleMLData, dml_params: Dict[str, Any])
                     {
                         "Learner g": learner_g_name,
                         "Learner m": learner_m_name,
+                        "Treatment Level": treatment_level,
                         "level": level,
                     }
                 )
@@ -120,7 +121,7 @@ def summarize_results(self):
         self.logger.info("Summarizing simulation results")
 
         # Group by parameter combinations
-        groupby_cols = ["Learner g", "Learner m", "level"]
+        groupby_cols = ["Learner g", "Learner m", "Treatment Level", "level"]
         aggregation_dict = {
             "Coverage": "mean",
             "CI Length": "mean",

Original file line number	Diff line number	Diff line change
`@@ -107,6 +107,7 @@ def run_single_rep(self, dml_data: dml.DoubleMLData, dml_params: Dict[str, Any])`
`107`	`107`	`{`
`108`	`108`	`"Learner g": learner_g_name,`
`109`	`109`	`"Learner m": learner_m_name,`
	`110`	`+ "Treatment Level": treatment_level,`
`110`	`111`	`"level": level,`
`111`	`112`	`}`
`112`	`113`	`)`
`@@ -120,7 +121,7 @@ def summarize_results(self):`
`120`	`121`	`self.logger.info("Summarizing simulation results")`
`121`	`122`
`122`	`123`	`# Group by parameter combinations`
`123`		`- groupby_cols = ["Learner g", "Learner m", "level"]`
	`124`	`+ groupby_cols = ["Learner g", "Learner m", "Treatment Level", "level"]`
`124`	`125`	`aggregation_dict = {`
`125`	`126`	`"Coverage": "mean",`
`126`	`127`	`"CI Length": "mean",`