├── .gitignore ├── README.md ├── camelyon-patient-distributions.csv ├── examples ├── __init__.py ├── algorithms │ ├── CDAN.py │ ├── DANN.py │ ├── ERM.py │ ├── IRM.py │ ├── algorithm.py │ ├── deepCORAL.py │ ├── group_algorithm.py │ ├── initializer.py │ └── single_model_algorithm.py ├── configs │ ├── algorithm.py │ ├── data_loader.py │ ├── datasets.py │ ├── model.py │ ├── scheduler.py │ ├── supported.py │ └── utils.py ├── data_augmentation │ ├── __init__.py │ ├── audio.py │ ├── batch_transform.py │ ├── bg_challenge.py │ ├── copy_paste.py │ ├── cutout.py │ ├── lisa.py │ ├── randaugment.py │ ├── random_crop.py │ ├── stain_color_jitter.py │ ├── transforms.py │ └── utils.py ├── losses.py ├── models │ ├── __init__.py │ ├── clip.py │ ├── domain_adversarial_network.py │ ├── initializer.py │ └── layers.py ├── optimizer.py ├── run_expt.py ├── scheduler.py ├── train.py └── utils.py ├── requirements.txt └── wilds ├── __init__.py ├── common ├── __init__.py ├── data_loaders.py ├── grouper.py ├── metrics │ ├── __init__.py │ ├── all_metrics.py │ ├── loss.py │ └── metric.py └── utils.py ├── datasets ├── __init__.py ├── amazon_dataset.py ├── archive │ ├── __init__.py │ ├── fmow_v1_0_dataset.py │ ├── iwildcam_v1_0_dataset.py │ └── poverty_v1_0_dataset.py ├── bdd100k_dataset.py ├── birdcalls_dataset.py ├── camelyon17_dataset.py ├── celebA_dataset.py ├── civilcomments_dataset.py ├── domainnet_dataset.py ├── download_utils.py ├── encode_dataset.py ├── fmow_dataset.py ├── globalwheat_dataset.py ├── iwildcam_dataset.py ├── ogbmolpcba_dataset.py ├── poverty_dataset.py ├── py150_dataset.py ├── rxrx1_dataset.py ├── sqf_dataset.py ├── unlabeled │ ├── __init__.py │ ├── amazon_unlabeled_dataset.py │ ├── camelyon17_unlabeled_dataset.py │ ├── civilcomments_unlabeled_dataset.py │ ├── domainnet_unlabeled_dataset.py │ ├── fmow_unlabeled_dataset.py │ ├── globalwheat_unlabeled_dataset.py │ ├── iwildcam_unlabeled_dataset.py │ ├── ogbmolpcba_unlabeled_dataset.py │ ├── poverty_unlabeled_dataset.py │ └── wilds_unlabeled_dataset.py ├── waterbirds_dataset.py ├── wilds_dataset.py └── yelp_dataset.py ├── download_datasets.py ├── get_dataset.py └── version.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/README.md -------------------------------------------------------------------------------- /camelyon-patient-distributions.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/camelyon-patient-distributions.csv -------------------------------------------------------------------------------- /examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/algorithms/CDAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/CDAN.py -------------------------------------------------------------------------------- /examples/algorithms/DANN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/DANN.py -------------------------------------------------------------------------------- /examples/algorithms/ERM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/ERM.py -------------------------------------------------------------------------------- /examples/algorithms/IRM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/IRM.py -------------------------------------------------------------------------------- /examples/algorithms/algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/algorithm.py -------------------------------------------------------------------------------- /examples/algorithms/deepCORAL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/deepCORAL.py -------------------------------------------------------------------------------- /examples/algorithms/group_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/group_algorithm.py -------------------------------------------------------------------------------- /examples/algorithms/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/initializer.py -------------------------------------------------------------------------------- /examples/algorithms/single_model_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/algorithms/single_model_algorithm.py -------------------------------------------------------------------------------- /examples/configs/algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/configs/algorithm.py -------------------------------------------------------------------------------- /examples/configs/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/configs/data_loader.py -------------------------------------------------------------------------------- /examples/configs/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/configs/datasets.py -------------------------------------------------------------------------------- /examples/configs/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/configs/model.py -------------------------------------------------------------------------------- /examples/configs/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/configs/scheduler.py -------------------------------------------------------------------------------- /examples/configs/supported.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/configs/supported.py -------------------------------------------------------------------------------- /examples/configs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/configs/utils.py -------------------------------------------------------------------------------- /examples/data_augmentation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/data_augmentation/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/audio.py -------------------------------------------------------------------------------- /examples/data_augmentation/batch_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/batch_transform.py -------------------------------------------------------------------------------- /examples/data_augmentation/bg_challenge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/bg_challenge.py -------------------------------------------------------------------------------- /examples/data_augmentation/copy_paste.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/copy_paste.py -------------------------------------------------------------------------------- /examples/data_augmentation/cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/cutout.py -------------------------------------------------------------------------------- /examples/data_augmentation/lisa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/lisa.py -------------------------------------------------------------------------------- /examples/data_augmentation/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/randaugment.py -------------------------------------------------------------------------------- /examples/data_augmentation/random_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/random_crop.py -------------------------------------------------------------------------------- /examples/data_augmentation/stain_color_jitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/stain_color_jitter.py -------------------------------------------------------------------------------- /examples/data_augmentation/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/transforms.py -------------------------------------------------------------------------------- /examples/data_augmentation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/data_augmentation/utils.py -------------------------------------------------------------------------------- /examples/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/losses.py -------------------------------------------------------------------------------- /examples/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/models/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/models/clip.py -------------------------------------------------------------------------------- /examples/models/domain_adversarial_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/models/domain_adversarial_network.py -------------------------------------------------------------------------------- /examples/models/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/models/initializer.py -------------------------------------------------------------------------------- /examples/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/models/layers.py -------------------------------------------------------------------------------- /examples/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/optimizer.py -------------------------------------------------------------------------------- /examples/run_expt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/run_expt.py -------------------------------------------------------------------------------- /examples/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/scheduler.py -------------------------------------------------------------------------------- /examples/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/train.py -------------------------------------------------------------------------------- /examples/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/examples/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | noisereduce 2 | librosa -------------------------------------------------------------------------------- /wilds/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/__init__.py -------------------------------------------------------------------------------- /wilds/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wilds/common/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/common/data_loaders.py -------------------------------------------------------------------------------- /wilds/common/grouper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/common/grouper.py -------------------------------------------------------------------------------- /wilds/common/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wilds/common/metrics/all_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/common/metrics/all_metrics.py -------------------------------------------------------------------------------- /wilds/common/metrics/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/common/metrics/loss.py -------------------------------------------------------------------------------- /wilds/common/metrics/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/common/metrics/metric.py -------------------------------------------------------------------------------- /wilds/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/common/utils.py -------------------------------------------------------------------------------- /wilds/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wilds/datasets/amazon_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/amazon_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/archive/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wilds/datasets/archive/fmow_v1_0_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/archive/fmow_v1_0_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/archive/iwildcam_v1_0_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/archive/iwildcam_v1_0_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/archive/poverty_v1_0_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/archive/poverty_v1_0_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/bdd100k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/bdd100k_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/birdcalls_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/birdcalls_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/camelyon17_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/camelyon17_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/celebA_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/celebA_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/civilcomments_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/civilcomments_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/domainnet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/domainnet_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/download_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/download_utils.py -------------------------------------------------------------------------------- /wilds/datasets/encode_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/encode_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/fmow_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/fmow_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/globalwheat_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/globalwheat_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/iwildcam_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/iwildcam_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/ogbmolpcba_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/ogbmolpcba_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/poverty_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/poverty_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/py150_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/py150_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/rxrx1_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/rxrx1_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/sqf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/sqf_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/amazon_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/amazon_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/camelyon17_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/camelyon17_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/civilcomments_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/civilcomments_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/domainnet_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/domainnet_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/fmow_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/fmow_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/globalwheat_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/globalwheat_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/iwildcam_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/iwildcam_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/ogbmolpcba_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/ogbmolpcba_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/poverty_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/poverty_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/unlabeled/wilds_unlabeled_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/unlabeled/wilds_unlabeled_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/waterbirds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/waterbirds_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/wilds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/wilds_dataset.py -------------------------------------------------------------------------------- /wilds/datasets/yelp_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/datasets/yelp_dataset.py -------------------------------------------------------------------------------- /wilds/download_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/download_datasets.py -------------------------------------------------------------------------------- /wilds/get_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/get_dataset.py -------------------------------------------------------------------------------- /wilds/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/i-gao/targeted-augs/HEAD/wilds/version.py --------------------------------------------------------------------------------