├── .gitignore ├── LICENSE.txt ├── README.md ├── drnet ├── __init__.py ├── apps │ ├── __init__.py │ ├── evaluate.py │ ├── load_db_icu.py │ ├── main.py │ ├── parameters.py │ ├── run_all_experiments.py │ └── util.py ├── data_access │ ├── __init__.py │ ├── batch_augmentation.py │ ├── data_access.py │ ├── generator.py │ ├── icu │ │ ├── __init__.py │ │ ├── data_access.py │ │ └── mimic3 │ │ │ ├── __init__.py │ │ │ └── data_access.py │ ├── ihdp │ │ ├── __init__.py │ │ ├── data_access.py │ │ └── generate_ihdp.R │ ├── jobs │ │ ├── __init__.py │ │ ├── data_access.py │ │ ├── jobs_DW_bin.test.npz │ │ └── jobs_DW_bin.train.npz │ ├── mahalanobis_batch.py │ ├── news │ │ ├── __init__.py │ │ └── data_access.py │ ├── patient_generator.py │ ├── propensity_batch.py │ ├── tcga │ │ ├── __init__.py │ │ ├── data_access.py │ │ ├── max_val.npy │ │ └── min_val.npy │ └── twins │ │ ├── __init__.py │ │ └── data_access.py ├── models │ ├── __init__.py │ ├── baselines │ │ ├── __init__.py │ │ ├── bart.py │ │ ├── baseline.py │ │ ├── causal_forest.py │ │ ├── cfr │ │ │ ├── __init__.py │ │ │ ├── cfr_net.py │ │ │ └── util.py │ │ ├── ganite.py │ │ ├── ganite_package │ │ │ ├── __init__.py │ │ │ ├── ganite_builder.py │ │ │ └── ganite_model.py │ │ ├── gaussian_process.py │ │ ├── gps.py │ │ ├── gradientboosted.py │ │ ├── knn.py │ │ ├── neural_network.py │ │ ├── ordinary_least_squares.py │ │ ├── psm.py │ │ ├── psm_pbm.py │ │ ├── random_forest.py │ │ └── tf_neural_network.py │ ├── benchmarks │ │ ├── __init__.py │ │ ├── benchmark.py │ │ ├── icu_benchmark.py │ │ ├── ihdp_benchmark.py │ │ ├── jobs_benchmark.py │ │ ├── news_benchmark.py │ │ ├── tcga_benchmark.py │ │ └── twins_benchmark.py │ ├── cf_early_stopping.py │ ├── distributions.py │ ├── exposure_metrics.py │ ├── model_builder.py │ ├── model_eval.py │ ├── model_factory.py │ ├── pehe_loss.py │ └── per_sample_dropout.py └── visualisation │ ├── confounding_plot.R │ ├── kappa_plot.R │ └── strata_plot.R ├── run_results.sh └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/README.md -------------------------------------------------------------------------------- /drnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/apps/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/apps/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/apps/evaluate.py -------------------------------------------------------------------------------- /drnet/apps/load_db_icu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/apps/load_db_icu.py -------------------------------------------------------------------------------- /drnet/apps/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/apps/main.py -------------------------------------------------------------------------------- /drnet/apps/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/apps/parameters.py -------------------------------------------------------------------------------- /drnet/apps/run_all_experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/apps/run_all_experiments.py -------------------------------------------------------------------------------- /drnet/apps/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/apps/util.py -------------------------------------------------------------------------------- /drnet/data_access/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/batch_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/batch_augmentation.py -------------------------------------------------------------------------------- /drnet/data_access/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/data_access.py -------------------------------------------------------------------------------- /drnet/data_access/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/generator.py -------------------------------------------------------------------------------- /drnet/data_access/icu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/icu/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/icu/data_access.py -------------------------------------------------------------------------------- /drnet/data_access/icu/mimic3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/icu/mimic3/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/icu/mimic3/data_access.py -------------------------------------------------------------------------------- /drnet/data_access/ihdp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/ihdp/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/ihdp/data_access.py -------------------------------------------------------------------------------- /drnet/data_access/ihdp/generate_ihdp.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/ihdp/generate_ihdp.R -------------------------------------------------------------------------------- /drnet/data_access/jobs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/jobs/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/jobs/data_access.py -------------------------------------------------------------------------------- /drnet/data_access/jobs/jobs_DW_bin.test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/jobs/jobs_DW_bin.test.npz -------------------------------------------------------------------------------- /drnet/data_access/jobs/jobs_DW_bin.train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/jobs/jobs_DW_bin.train.npz -------------------------------------------------------------------------------- /drnet/data_access/mahalanobis_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/mahalanobis_batch.py -------------------------------------------------------------------------------- /drnet/data_access/news/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/news/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/news/data_access.py -------------------------------------------------------------------------------- /drnet/data_access/patient_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/patient_generator.py -------------------------------------------------------------------------------- /drnet/data_access/propensity_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/propensity_batch.py -------------------------------------------------------------------------------- /drnet/data_access/tcga/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/tcga/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/tcga/data_access.py -------------------------------------------------------------------------------- /drnet/data_access/tcga/max_val.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/tcga/max_val.npy -------------------------------------------------------------------------------- /drnet/data_access/tcga/min_val.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/tcga/min_val.npy -------------------------------------------------------------------------------- /drnet/data_access/twins/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/data_access/twins/data_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/data_access/twins/data_access.py -------------------------------------------------------------------------------- /drnet/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/models/baselines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/models/baselines/bart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/bart.py -------------------------------------------------------------------------------- /drnet/models/baselines/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/baseline.py -------------------------------------------------------------------------------- /drnet/models/baselines/causal_forest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/causal_forest.py -------------------------------------------------------------------------------- /drnet/models/baselines/cfr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/models/baselines/cfr/cfr_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/cfr/cfr_net.py -------------------------------------------------------------------------------- /drnet/models/baselines/cfr/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/cfr/util.py -------------------------------------------------------------------------------- /drnet/models/baselines/ganite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/ganite.py -------------------------------------------------------------------------------- /drnet/models/baselines/ganite_package/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/models/baselines/ganite_package/ganite_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/ganite_package/ganite_builder.py -------------------------------------------------------------------------------- /drnet/models/baselines/ganite_package/ganite_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/ganite_package/ganite_model.py -------------------------------------------------------------------------------- /drnet/models/baselines/gaussian_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/gaussian_process.py -------------------------------------------------------------------------------- /drnet/models/baselines/gps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/gps.py -------------------------------------------------------------------------------- /drnet/models/baselines/gradientboosted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/gradientboosted.py -------------------------------------------------------------------------------- /drnet/models/baselines/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/knn.py -------------------------------------------------------------------------------- /drnet/models/baselines/neural_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/neural_network.py -------------------------------------------------------------------------------- /drnet/models/baselines/ordinary_least_squares.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/ordinary_least_squares.py -------------------------------------------------------------------------------- /drnet/models/baselines/psm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/psm.py -------------------------------------------------------------------------------- /drnet/models/baselines/psm_pbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/psm_pbm.py -------------------------------------------------------------------------------- /drnet/models/baselines/random_forest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/random_forest.py -------------------------------------------------------------------------------- /drnet/models/baselines/tf_neural_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/baselines/tf_neural_network.py -------------------------------------------------------------------------------- /drnet/models/benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /drnet/models/benchmarks/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/benchmarks/benchmark.py -------------------------------------------------------------------------------- /drnet/models/benchmarks/icu_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/benchmarks/icu_benchmark.py -------------------------------------------------------------------------------- /drnet/models/benchmarks/ihdp_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/benchmarks/ihdp_benchmark.py -------------------------------------------------------------------------------- /drnet/models/benchmarks/jobs_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/benchmarks/jobs_benchmark.py -------------------------------------------------------------------------------- /drnet/models/benchmarks/news_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/benchmarks/news_benchmark.py -------------------------------------------------------------------------------- /drnet/models/benchmarks/tcga_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/benchmarks/tcga_benchmark.py -------------------------------------------------------------------------------- /drnet/models/benchmarks/twins_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/benchmarks/twins_benchmark.py -------------------------------------------------------------------------------- /drnet/models/cf_early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/cf_early_stopping.py -------------------------------------------------------------------------------- /drnet/models/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/distributions.py -------------------------------------------------------------------------------- /drnet/models/exposure_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/exposure_metrics.py -------------------------------------------------------------------------------- /drnet/models/model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/model_builder.py -------------------------------------------------------------------------------- /drnet/models/model_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/model_eval.py -------------------------------------------------------------------------------- /drnet/models/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/model_factory.py -------------------------------------------------------------------------------- /drnet/models/pehe_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/pehe_loss.py -------------------------------------------------------------------------------- /drnet/models/per_sample_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/models/per_sample_dropout.py -------------------------------------------------------------------------------- /drnet/visualisation/confounding_plot.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/visualisation/confounding_plot.R -------------------------------------------------------------------------------- /drnet/visualisation/kappa_plot.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/visualisation/kappa_plot.R -------------------------------------------------------------------------------- /drnet/visualisation/strata_plot.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/drnet/visualisation/strata_plot.R -------------------------------------------------------------------------------- /run_results.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/run_results.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/d909b/drnet/HEAD/setup.py --------------------------------------------------------------------------------