survival-datasets
A simple data loader for the most common datasets in Survival Analysis. Currently the following are included:
- Veterans Lung Cancer (https://scikit-survival.readthedocs.io/en/stable/api/generated/sksurv.datasets.load_veterans_lung_cancer.html)
- German Breast Cancer Study Group (GBSG2) (https://scikit-survival.readthedocs.io/en/stable/api/generated/sksurv.datasets.load_gbsg2.html)
- AIDS dataset (https://scikit-survival.readthedocs.io/en/stable/api/generated/sksurv.datasets.load_aids.html)
- NHANES (https://shap.readthedocs.io/en/latest/generated/shap.datasets.nhanesi.html)
- SUPPORT Study to Understand Prognoses Preferences Outcomes and Risks of Treatment (from DeepSurv paper, https://arxiv.org/abs/1606.00931)
- METABRIC The Molecular Taxonomy of Breast Cancer International Consortium (from DeepSurv paper, https://arxiv.org/abs/1606.00931)
- WHAS500 Worcester Heart Attack Study (https://scikit-survival.readthedocs.io/en/stable/api/datasets.html)
- FLCHAIN (https://scikit-survival.readthedocs.io/en/stable/api/datasets.html)
- SEER (from Kaggle, https://www.kaggle.com/code/jnegrini/breast-cancer-dataset)
Requirements
- Python 3.8 or later
- scikit-survival 0.17.2 or later
- pandas 1.4.3 or later
- numpy 1.22.4 or later
- shap 0.41 or later
- pyarrow 11.0 or later
Installation
Simply install via pip:
pip install survival-datasets
Examples
Import the datasets module from the package and load your dataset of choice:
from survdata import datasets
if __name__ == "__main__":
X, y = datasets.load_seer_dataset()
1 maintainer
Metadata
1 maintainer
Release files for survival-datasets 0.1.5
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| survival-datasets-0.1.5.tar.gz | 263.1 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| survival_datasets-0.1.5-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 526.6 kB
Release files / survival-datasets-0.1.5.tar.gz
| Download URL | survival-datasets-0.1.5.tar.gz |
|---|---|
| Size | 263.1 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
41ce2ce91ef84f111d2dd34d8d579cbc72489600d7a3b15ca22ed32437a35017
|
|
BLAKE2b-256 checksum How to use checksums |
ebba72b67768f01b31cd4098c03d1c9545572ddb33adaa8202b41bdf2c97e8b8
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.8.16
|
Release files / survival_datasets-0.1.5-py3-none-any.whl
| Download URL | survival_datasets-0.1.5-py3-none-any.whl |
|---|---|
| Size | 263.4 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
87f5720816eb2e385b06f6815ff1d1bd8f27efd63b059d3ad51c222467b4ff68
|
|
BLAKE2b-256 checksum How to use checksums |
eb3b7f7fbfbedca2c991fe122eac753dee27450a2c54a8c9bcf06f94a9992bf2
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.8.16
|
