Skip to content

Creating Pooled datasets

Robin van de Water edited this page Aug 30, 2023 · 1 revision

In icu_benchmarks/data/pooling.py you can find functions to generate pooled cohorts dynamically. The following is a code snippet to do perform this operation:

Datasets = [PooledDataset.aumc_eicu_miiv, PooledDataset.hirid_eicu_miiv, PooledDataset.aumc_hirid_eicu, PooledDataset.aumc_hirid_miiv, PooledDataset.aumc_hirid_eicu_miiv]
for item in Datasets:
        generate_pooled_data(Path(r'cohorts\mortality24'), vars=vars, datasets=item, file_names=file_names, seed=seed, runmode=runmode)
Clone this wiki locally