├── CausalM ├── BERT │ ├── bert-base-cased-config.json │ ├── bert_pos_tagger.py │ ├── bert_text_classifier.py │ ├── bert_text_dataset.py │ └── lm_finetune │ │ ├── bert_mlm_finetune.py │ │ └── grad_reverse_layer.py ├── LICENSE ├── POMS_GendeRace │ ├── README.md │ ├── lm_finetune │ │ ├── BERT │ │ │ ├── bert_pos_tagger.py │ │ │ ├── bert_text_classifier.py │ │ │ ├── bert_text_dataset.py │ │ │ └── lm_finetune │ │ │ │ ├── bert_mlm_finetune.py │ │ │ │ └── grad_reverse_layer.py │ │ ├── bert_genderace_finetune.py │ │ ├── constants.py │ │ ├── datasets │ │ │ └── utils.py │ │ ├── genderace_finetune_on_pregenerated.py │ │ ├── mlm_finetune_on_pregenerated.py │ │ ├── pregenerate_training_data.py │ │ └── utils.py │ └── pipeline │ │ ├── BERT │ │ ├── bert_pos_tagger.py │ │ ├── bert_text_classifier.py │ │ ├── bert_text_dataset.py │ │ └── lm_finetune │ │ │ ├── bert_mlm_finetune.py │ │ │ └── grad_reverse_layer.py │ │ ├── TransTEE │ │ ├── DisCri.py │ │ ├── dosage.py │ │ ├── dynamic_attention.py │ │ ├── dynamic_net.py │ │ ├── trans_ci.py │ │ ├── transformers.py │ │ └── transtee.py │ │ ├── constants.py │ │ ├── datasets │ │ └── utils.py │ │ ├── predict.py │ │ ├── training.py │ │ └── utils.py ├── README.md ├── _layouts │ └── default.html ├── causalm_gpu_env.yml ├── constants.py ├── datasets │ └── utils.py └── utils.py ├── Continuous ├── README.md ├── data │ ├── data.py │ ├── ihdp.py │ ├── ihdp_generate_data.py │ ├── news_generate_data.py │ ├── news_process.py │ ├── plot_curve.py │ ├── simu1.py │ └── simu1_generate_data.py ├── dataset │ ├── ihdp │ │ └── ihdp.csv │ └── news │ │ └── news_p.csv ├── main.py ├── main_batch.py ├── main_batch_ihdpv2.py ├── main_batch_newsv2.py ├── models │ ├── DisCri.py │ ├── dosage.py │ ├── dynamic_attention.py │ ├── dynamic_net.py │ ├── itermidate_models.py │ ├── trans_ci.py │ └── transformers.py ├── utils │ ├── eval.py │ ├── eval_ihdp.py │ └── tsne.py └── visualize.py ├── Dosage ├── DRNet.py ├── README.md ├── SCIGAN.py ├── TransTEE.py ├── data_simulation.py ├── process_result.py ├── test_DRNet.py ├── test_SCIGAN.py ├── test_TransTEE.py └── utils │ ├── ardf_curve.py │ ├── draw_results.py │ ├── evaluation_torch.py │ ├── evaluation_utils.py │ ├── mlp.py │ ├── model_utils.py │ ├── trans_ci.py │ ├── transformers.py │ ├── treatment_bias.py │ └── utils.py ├── LICENSE ├── README.md ├── Structured ├── README.md ├── configs │ ├── generate_data │ │ ├── sw.py │ │ └── tcga.py │ ├── run_model │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-36.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── sw.cpython-36.pyc │ │ │ ├── sw.cpython-38.pyc │ │ │ ├── tcga.cpython-36.pyc │ │ │ └── tcga.cpython-38.pyc │ │ ├── sw.py │ │ └── tcga.py │ └── sweeps │ │ └── sw │ │ ├── cat.yaml │ │ ├── gnn.yaml │ │ ├── graphite.yaml │ │ └── sin.yaml ├── data │ ├── dataset.py │ ├── tcga │ │ ├── qm9_tcga_simulation.py │ │ └── smiles_processing.py │ └── utils.py ├── experiments │ ├── early_stopping.py │ ├── evaluate.py │ ├── io.py │ ├── logger.py │ ├── train.py │ └── utils.py ├── generate_data.py ├── models │ ├── building_blocks │ │ ├── como.py │ │ ├── covariates_feature_extractor.py │ │ ├── gnn.py │ │ ├── hsic_utils.py │ │ ├── mlp.py │ │ ├── neural_network.py │ │ ├── outcome_model.py │ │ ├── propensity_feature_extractor.py │ │ ├── transformers.py │ │ ├── treatment_feature_extractor.py │ │ ├── utils.py │ │ └── zero_baseline.py │ ├── cat.py │ ├── gnn.py │ ├── graphite.py │ ├── sin.py │ └── transtee.py ├── requirements.txt ├── run_hyperparameter_sweeping.py ├── run_model_training.py ├── run_unseen_treatment_update.py └── simulation │ ├── data_generator.py │ ├── outcome_generators.py │ ├── small_world │ ├── outcome_simulator.py │ └── treatment_generator.py │ ├── tcga │ ├── data_generator.py │ └── outcome_simulator.py │ ├── treatment_assignment.py │ ├── treatment_generators.py │ ├── unit_generators.py │ └── utils.py └── model.png /CausalM/BERT/bert-base-cased-config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/BERT/bert-base-cased-config.json -------------------------------------------------------------------------------- /CausalM/BERT/bert_pos_tagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/BERT/bert_pos_tagger.py -------------------------------------------------------------------------------- /CausalM/BERT/bert_text_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/BERT/bert_text_classifier.py -------------------------------------------------------------------------------- /CausalM/BERT/bert_text_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/BERT/bert_text_dataset.py -------------------------------------------------------------------------------- /CausalM/BERT/lm_finetune/bert_mlm_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/BERT/lm_finetune/bert_mlm_finetune.py -------------------------------------------------------------------------------- /CausalM/BERT/lm_finetune/grad_reverse_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/BERT/lm_finetune/grad_reverse_layer.py -------------------------------------------------------------------------------- /CausalM/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/LICENSE -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/README.md -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/BERT/bert_pos_tagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/BERT/bert_pos_tagger.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/BERT/bert_text_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/BERT/bert_text_classifier.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/BERT/bert_text_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/BERT/bert_text_dataset.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/BERT/lm_finetune/bert_mlm_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/BERT/lm_finetune/bert_mlm_finetune.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/BERT/lm_finetune/grad_reverse_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/BERT/lm_finetune/grad_reverse_layer.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/bert_genderace_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/bert_genderace_finetune.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/constants.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/datasets/utils.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/genderace_finetune_on_pregenerated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/genderace_finetune_on_pregenerated.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/mlm_finetune_on_pregenerated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/mlm_finetune_on_pregenerated.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/pregenerate_training_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/pregenerate_training_data.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/lm_finetune/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/lm_finetune/utils.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/BERT/bert_pos_tagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/BERT/bert_pos_tagger.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/BERT/bert_text_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/BERT/bert_text_classifier.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/BERT/bert_text_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/BERT/bert_text_dataset.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/BERT/lm_finetune/bert_mlm_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/BERT/lm_finetune/bert_mlm_finetune.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/BERT/lm_finetune/grad_reverse_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/BERT/lm_finetune/grad_reverse_layer.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/TransTEE/dosage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/dosage.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/TransTEE/dynamic_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/dynamic_attention.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/TransTEE/dynamic_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/dynamic_net.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/TransTEE/trans_ci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/trans_ci.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/TransTEE/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/transformers.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/TransTEE/transtee.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/transtee.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/constants.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/datasets/utils.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/predict.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/training.py -------------------------------------------------------------------------------- /CausalM/POMS_GendeRace/pipeline/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/POMS_GendeRace/pipeline/utils.py -------------------------------------------------------------------------------- /CausalM/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/README.md -------------------------------------------------------------------------------- /CausalM/_layouts/default.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/_layouts/default.html -------------------------------------------------------------------------------- /CausalM/causalm_gpu_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/causalm_gpu_env.yml -------------------------------------------------------------------------------- /CausalM/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/constants.py -------------------------------------------------------------------------------- /CausalM/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/datasets/utils.py -------------------------------------------------------------------------------- /CausalM/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/CausalM/utils.py -------------------------------------------------------------------------------- /Continuous/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/README.md -------------------------------------------------------------------------------- /Continuous/data/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/data.py -------------------------------------------------------------------------------- /Continuous/data/ihdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/ihdp.py -------------------------------------------------------------------------------- /Continuous/data/ihdp_generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/ihdp_generate_data.py -------------------------------------------------------------------------------- /Continuous/data/news_generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/news_generate_data.py -------------------------------------------------------------------------------- /Continuous/data/news_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/news_process.py -------------------------------------------------------------------------------- /Continuous/data/plot_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/plot_curve.py -------------------------------------------------------------------------------- /Continuous/data/simu1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/simu1.py -------------------------------------------------------------------------------- /Continuous/data/simu1_generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/data/simu1_generate_data.py -------------------------------------------------------------------------------- /Continuous/dataset/ihdp/ihdp.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/dataset/ihdp/ihdp.csv -------------------------------------------------------------------------------- /Continuous/dataset/news/news_p.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/dataset/news/news_p.csv -------------------------------------------------------------------------------- /Continuous/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/main.py -------------------------------------------------------------------------------- /Continuous/main_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/main_batch.py -------------------------------------------------------------------------------- /Continuous/main_batch_ihdpv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/main_batch_ihdpv2.py -------------------------------------------------------------------------------- /Continuous/main_batch_newsv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/main_batch_newsv2.py -------------------------------------------------------------------------------- /Continuous/models/DisCri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/models/DisCri.py -------------------------------------------------------------------------------- /Continuous/models/dosage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/models/dosage.py -------------------------------------------------------------------------------- /Continuous/models/dynamic_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/models/dynamic_attention.py -------------------------------------------------------------------------------- /Continuous/models/dynamic_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/models/dynamic_net.py -------------------------------------------------------------------------------- /Continuous/models/itermidate_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/models/itermidate_models.py -------------------------------------------------------------------------------- /Continuous/models/trans_ci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/models/trans_ci.py -------------------------------------------------------------------------------- /Continuous/models/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/models/transformers.py -------------------------------------------------------------------------------- /Continuous/utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/utils/eval.py -------------------------------------------------------------------------------- /Continuous/utils/eval_ihdp.py: -------------------------------------------------------------------------------- 1 | import numpy as np 2 | 3 | 4 | -------------------------------------------------------------------------------- /Continuous/utils/tsne.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/utils/tsne.py -------------------------------------------------------------------------------- /Continuous/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Continuous/visualize.py -------------------------------------------------------------------------------- /Dosage/DRNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/DRNet.py -------------------------------------------------------------------------------- /Dosage/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/README.md -------------------------------------------------------------------------------- /Dosage/SCIGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/SCIGAN.py -------------------------------------------------------------------------------- /Dosage/TransTEE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/TransTEE.py -------------------------------------------------------------------------------- /Dosage/data_simulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/data_simulation.py -------------------------------------------------------------------------------- /Dosage/process_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/process_result.py -------------------------------------------------------------------------------- /Dosage/test_DRNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/test_DRNet.py -------------------------------------------------------------------------------- /Dosage/test_SCIGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/test_SCIGAN.py -------------------------------------------------------------------------------- /Dosage/test_TransTEE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/test_TransTEE.py -------------------------------------------------------------------------------- /Dosage/utils/ardf_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/ardf_curve.py -------------------------------------------------------------------------------- /Dosage/utils/draw_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/draw_results.py -------------------------------------------------------------------------------- /Dosage/utils/evaluation_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/evaluation_torch.py -------------------------------------------------------------------------------- /Dosage/utils/evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/evaluation_utils.py -------------------------------------------------------------------------------- /Dosage/utils/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/mlp.py -------------------------------------------------------------------------------- /Dosage/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/model_utils.py -------------------------------------------------------------------------------- /Dosage/utils/trans_ci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/trans_ci.py -------------------------------------------------------------------------------- /Dosage/utils/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/transformers.py -------------------------------------------------------------------------------- /Dosage/utils/treatment_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/treatment_bias.py -------------------------------------------------------------------------------- /Dosage/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Dosage/utils/utils.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/README.md -------------------------------------------------------------------------------- /Structured/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/README.md -------------------------------------------------------------------------------- /Structured/configs/generate_data/sw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/generate_data/sw.py -------------------------------------------------------------------------------- /Structured/configs/generate_data/tcga.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/generate_data/tcga.py -------------------------------------------------------------------------------- /Structured/configs/run_model/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /Structured/configs/run_model/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /Structured/configs/run_model/__pycache__/sw.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/__pycache__/sw.cpython-36.pyc -------------------------------------------------------------------------------- /Structured/configs/run_model/__pycache__/sw.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/__pycache__/sw.cpython-38.pyc -------------------------------------------------------------------------------- /Structured/configs/run_model/__pycache__/tcga.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/__pycache__/tcga.cpython-36.pyc -------------------------------------------------------------------------------- /Structured/configs/run_model/__pycache__/tcga.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/__pycache__/tcga.cpython-38.pyc -------------------------------------------------------------------------------- /Structured/configs/run_model/sw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/sw.py -------------------------------------------------------------------------------- /Structured/configs/run_model/tcga.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/run_model/tcga.py -------------------------------------------------------------------------------- /Structured/configs/sweeps/sw/cat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/sweeps/sw/cat.yaml -------------------------------------------------------------------------------- /Structured/configs/sweeps/sw/gnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/sweeps/sw/gnn.yaml -------------------------------------------------------------------------------- /Structured/configs/sweeps/sw/graphite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/sweeps/sw/graphite.yaml -------------------------------------------------------------------------------- /Structured/configs/sweeps/sw/sin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/configs/sweeps/sw/sin.yaml -------------------------------------------------------------------------------- /Structured/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/data/dataset.py -------------------------------------------------------------------------------- /Structured/data/tcga/qm9_tcga_simulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/data/tcga/qm9_tcga_simulation.py -------------------------------------------------------------------------------- /Structured/data/tcga/smiles_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/data/tcga/smiles_processing.py -------------------------------------------------------------------------------- /Structured/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/data/utils.py -------------------------------------------------------------------------------- /Structured/experiments/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/experiments/early_stopping.py -------------------------------------------------------------------------------- /Structured/experiments/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/experiments/evaluate.py -------------------------------------------------------------------------------- /Structured/experiments/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/experiments/io.py -------------------------------------------------------------------------------- /Structured/experiments/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/experiments/logger.py -------------------------------------------------------------------------------- /Structured/experiments/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/experiments/train.py -------------------------------------------------------------------------------- /Structured/experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/experiments/utils.py -------------------------------------------------------------------------------- /Structured/generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/generate_data.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/como.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/como.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/covariates_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/covariates_feature_extractor.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/gnn.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/hsic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/hsic_utils.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/mlp.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/neural_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/neural_network.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/outcome_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/outcome_model.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/propensity_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/propensity_feature_extractor.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/transformers.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/treatment_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/treatment_feature_extractor.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/utils.py -------------------------------------------------------------------------------- /Structured/models/building_blocks/zero_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/building_blocks/zero_baseline.py -------------------------------------------------------------------------------- /Structured/models/cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/cat.py -------------------------------------------------------------------------------- /Structured/models/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/gnn.py -------------------------------------------------------------------------------- /Structured/models/graphite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/graphite.py -------------------------------------------------------------------------------- /Structured/models/sin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/sin.py -------------------------------------------------------------------------------- /Structured/models/transtee.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/models/transtee.py -------------------------------------------------------------------------------- /Structured/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/requirements.txt -------------------------------------------------------------------------------- /Structured/run_hyperparameter_sweeping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/run_hyperparameter_sweeping.py -------------------------------------------------------------------------------- /Structured/run_model_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/run_model_training.py -------------------------------------------------------------------------------- /Structured/run_unseen_treatment_update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/run_unseen_treatment_update.py -------------------------------------------------------------------------------- /Structured/simulation/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/data_generator.py -------------------------------------------------------------------------------- /Structured/simulation/outcome_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/outcome_generators.py -------------------------------------------------------------------------------- /Structured/simulation/small_world/outcome_simulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/small_world/outcome_simulator.py -------------------------------------------------------------------------------- /Structured/simulation/small_world/treatment_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/small_world/treatment_generator.py -------------------------------------------------------------------------------- /Structured/simulation/tcga/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/tcga/data_generator.py -------------------------------------------------------------------------------- /Structured/simulation/tcga/outcome_simulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/tcga/outcome_simulator.py -------------------------------------------------------------------------------- /Structured/simulation/treatment_assignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/treatment_assignment.py -------------------------------------------------------------------------------- /Structured/simulation/treatment_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/treatment_generators.py -------------------------------------------------------------------------------- /Structured/simulation/unit_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/unit_generators.py -------------------------------------------------------------------------------- /Structured/simulation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/Structured/simulation/utils.py -------------------------------------------------------------------------------- /model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hlzhang109/TransTEE/HEAD/model.png --------------------------------------------------------------------------------