search

BiomedSciAI / causallib

A Python package for modular causal inference analysis and model evaluations
Apache License 2.0
728 stars 97 forks source link

Fix individual survival curves estimation bug #56

Closed ehudkr closed 1 year ago

ehudkr commented 1 year ago

Several bugs found and fixed in StandardizedSurvival:

  1. Multi-index of treatment over observations was made explicit.
  2. In stratify=True, prediction is done on the entire dataset
  3. In stratify=False, prediction is done when forcing same treatment assignment for all.

All the above probably made results closer to the observed (unadjusted) survival curves.