Add zero weight for censored

Quantco · Jul 12, 2024 · 2a3cd76 · 2a3cd76
1 parent ae2d825
commit 2a3cd76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/examples/example_survival.ipynb b/docs/examples/example_survival.ipynb
@@ -163,6 +163,7 @@
     "    inverse_prob_censoring_weight = pd.Series(\n",
     "        index=outcome.index, data=inverse_prob_censoring_weight.values\n",
     "    )\n",
+    "    inverse_prob_censoring_weight[censored.astype(bool)] = 0\n",
     "\n",
     "    return inverse_prob_censoring_weight\n",
     "\n",