Avoid using 'Est' as a short hand for 'Estimator'

soda-inria · Jul 2, 2024 · 94e7397 · 94e7397
1 parent 527f88d
commit 94e7397
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/hazardous/_ipcw.py b/hazardous/_ipcw.py
@@ -145,7 +145,7 @@ def compute_censoring_survival_proba(self, times, X=None, ipcw_training=False):
         return self.censoring_survival_func_(times)
 
 
-class AlternatingCensoringEst(BaseIPCW):
+class AlternatingCensoringEstimator(BaseIPCW):
     """IPCW estimator for Debiased Gradient Boosting Incidence.
 
     Predict :math:`\hat{G}(t | X = x) = p(C > t | X = x)` using

diff --git a/hazardous/_survival_boost.py b/hazardous/_survival_boost.py
@@ -4,7 +4,7 @@
 from sklearn.utils.validation import check_array, check_random_state
 from tqdm import tqdm
 
-from ._ipcw import AlternatingCensoringEst
+from ._ipcw import AlternatingCensoringEstimator
 from .metrics._brier_score import (
     IncidenceScoreComputer,
     integrated_brier_score_incidence,
@@ -219,7 +219,7 @@ def fit(self, X, y, times=None):
             self.time_grid_.sort()
 
         if self.ipcw_est is None:
-            ipcw_est = AlternatingCensoringEst(incidence_est=self.estimator_)
+            ipcw_est = AlternatingCensoringEstimator(incidence_est=self.estimator_)
         else:
             ipcw_est = self.ipcw_est
 
@@ -263,7 +263,7 @@ def fit(self, X, y, times=None):
                 )
 
             if (idx_iter % self.n_iter_before_feedback == 0) and isinstance(
-                ipcw_est, AlternatingCensoringEst
+                ipcw_est, AlternatingCensoringEstimator
             ):
                 self.weighted_targets_.fit(X)