Merge pull request #1440 from rocreguant/master

bugfix: 'last_iteration' not initialized
CamDavidsonPilon · Jul 17, 2022 · f156b7f · f156b7f
2 parents f842388 + e3aecec
commit f156b7f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lifelines/fitters/aalen_additive_fitter.py b/lifelines/fitters/aalen_additive_fitter.py
@@ -223,6 +223,7 @@ def _fit_model_to_data_batch(self, X, T, E, weights, show_progress):
         variance_hazards_ = np.zeros((n_deaths, d))
         v = np.zeros(d)
         start = time.time()
+        last_iteration = 0
 
         W = np.sqrt(weights)
         X = W[:, None] * X