From 362ecb139a85a2ad1f9dcce0baff1e50b81507a4 Mon Sep 17 00:00:00 2001 From: Michael Foster Date: Thu, 22 Aug 2024 10:24:14 +0100 Subject: [PATCH 1/2] fixed alpha bug --- causal_testing/testing/estimators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/causal_testing/testing/estimators.py b/causal_testing/testing/estimators.py index 3920e1f2..ffe76387 100644 --- a/causal_testing/testing/estimators.py +++ b/causal_testing/testing/estimators.py @@ -823,7 +823,7 @@ def estimate_hazard_ratio(self): # IPCW step 4: Use these weights in a weighted analysis of the outcome model # Estimate the KM graph and IPCW hazard ratio using Cox regression. - cox_ph = CoxPHFitter() + cox_ph = CoxPHFitter(alpha=self.alpha) cox_ph.fit( df=preprocessed_data, duration_col="tout", From 4d1176458414b98ed4a473ad093ff26a1fb22d47 Mon Sep 17 00:00:00 2001 From: Michael Foster Date: Thu, 22 Aug 2024 10:26:59 +0100 Subject: [PATCH 2/2] black --- causal_testing/surrogate/causal_surrogate_assisted.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/causal_testing/surrogate/causal_surrogate_assisted.py b/causal_testing/surrogate/causal_surrogate_assisted.py index c30b2086..c67d4e4b 100644 --- a/causal_testing/surrogate/causal_surrogate_assisted.py +++ b/causal_testing/surrogate/causal_surrogate_assisted.py @@ -9,6 +9,7 @@ from causal_testing.testing.base_test_case import BaseTestCase from causal_testing.testing.estimators import CubicSplineRegressionEstimator + @dataclass class SimulationResult: """Data class holding the data and result metadata of a simulation""" @@ -19,7 +20,7 @@ class SimulationResult: def to_dataframe(self) -> pd.DataFrame: """Convert the simulation result data to a pandas DataFrame""" - data_as_lists = {k: v if isinstance(v, list) else [v] for k,v in self.data.items()} + data_as_lists = {k: v if isinstance(v, list) else [v] for k, v in self.data.items()} return pd.DataFrame(data_as_lists)