Skip to content

Commit fc2bb56

Browse files
eigenfootwiecki
authored andcommitted
Censored data and Weibull AFT model (#3095)
* Port of jazzmuesli PR * Add explanation and docstrings, rename notebook * Finished censored data notebook * Rename aft notebook to weibull aft * Revert "Finished censored data notebook" This reverts commit 74656e3. * Revert accidental change * Remove censored_data.py * Revert "Revert "Finished censored data notebook"" This reverts commit 4c6faab. * Touch up censored_data.ipynb * Add survival analysis session to examples * Add parametric survival notebook * Update doc session * First pass at AFT notebook * Second pass at weibull aft notebook * Change all Bernoullis to Potentials * Update doc to point to weibull aft nb * Include some more explanation * Touch ups * Add line to release notes * Correct Bernoulli to Potential
1 parent 6631265 commit fc2bb56

File tree

7 files changed

+2042
-134
lines changed

7 files changed

+2042
-134
lines changed

RELEASE-NOTES.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
### New features
88

9+
- Add documentation section on survival analysis and censored data models
910
- Add `check_test_point` method to `pm.Model`
1011
- Add `Ordered` Transformation and `OrderedLogistic` distribution
1112
- Add `Chain` transformation

docs/source/api/bounds.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ Caveats
5454

5555
* Bounds cannot be given to variables that are ``observed``. To model
5656
truncated data, use a :func:`~pymc3.model.Potential` in combination with a cumulative
57-
probability function. See `this example <https://github.com/pymc-devs/pymc3/blob/master/pymc3/examples/censored_data.py>`_.
57+
probability function. See `this example notebook <https://github.com/pymc-devs/pymc3/blob/master/docs/source/notebooks/weibull_aft.ipynb>`_.
5858

5959
* The automatic transformation applied to continuous distributions results in
6060
an unnormalized probability distribution. This doesn't effect inference

docs/source/examples.rst

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@ Applied
2828
notebooks/stochastic_volatility.ipynb
2929
notebooks/probabilistic_matrix_factorization.ipynb
3030
notebooks/rugby_analytics.ipynb
31-
notebooks/survival_analysis.ipynb
3231
notebooks/dawid-skene.ipynb
3332

3433
GLM
@@ -46,6 +45,15 @@ GLM
4645
notebooks/GLM-negative-binomial-regression.ipynb
4746
notebooks/GLM-hierarchical-binominal-model.ipynb
4847

48+
Survival Analysis
49+
=================
50+
51+
.. toctree::
52+
notebooks/survival_analysis.ipynb
53+
notebooks/bayes_param_survival_pymc3.ipynb
54+
notebooks/weibull_aft.ipynb
55+
notebooks/censored_data.ipynb
56+
4957
Gaussian Processes
5058
==================
5159

@@ -80,7 +88,6 @@ Variational Inference
8088
notebooks/empirical-approx-overview.ipynb
8189
notebooks/normalizing_flows_overview.ipynb
8290

83-
8491
Stochastic Gradient
8592
===================
8693

docs/source/notebooks/bayes_param_survival_pymc3.ipynb

Lines changed: 993 additions & 0 deletions
Large diffs are not rendered by default.

docs/source/notebooks/censored_data.ipynb

Lines changed: 407 additions & 0 deletions
Large diffs are not rendered by default.

docs/source/notebooks/weibull_aft.ipynb

Lines changed: 631 additions & 0 deletions
Large diffs are not rendered by default.

pymc3/examples/censored_data.py

Lines changed: 0 additions & 131 deletions
This file was deleted.

0 commit comments

Comments
 (0)