├── LICENSE ├── README.md ├── analysis_utils.py ├── data ├── celebA_dataset.py ├── confounder_dataset.py ├── confounder_utils.py ├── cub_dataset.py ├── data.py ├── dro_dataset.py ├── label_shift_utils.py ├── multinli_dataset.py ├── torchvision_datasets.py └── utils.py ├── dataset_metadata └── multinli │ ├── metadata_preset.csv │ └── metadata_random.csv ├── dataset_scripts ├── dataset_utils.py ├── generate_multinli.py └── generate_waterbirds.py ├── loss.py ├── models.py ├── run_expt.py ├── train.py ├── utils.py └── utils_glue.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/README.md -------------------------------------------------------------------------------- /analysis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/analysis_utils.py -------------------------------------------------------------------------------- /data/celebA_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/celebA_dataset.py -------------------------------------------------------------------------------- /data/confounder_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/confounder_dataset.py -------------------------------------------------------------------------------- /data/confounder_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/confounder_utils.py -------------------------------------------------------------------------------- /data/cub_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/cub_dataset.py -------------------------------------------------------------------------------- /data/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/data.py -------------------------------------------------------------------------------- /data/dro_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/dro_dataset.py -------------------------------------------------------------------------------- /data/label_shift_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/label_shift_utils.py -------------------------------------------------------------------------------- /data/multinli_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/multinli_dataset.py -------------------------------------------------------------------------------- /data/torchvision_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/torchvision_datasets.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/data/utils.py -------------------------------------------------------------------------------- /dataset_metadata/multinli/metadata_preset.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/dataset_metadata/multinli/metadata_preset.csv -------------------------------------------------------------------------------- /dataset_metadata/multinli/metadata_random.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/dataset_metadata/multinli/metadata_random.csv -------------------------------------------------------------------------------- /dataset_scripts/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/dataset_scripts/dataset_utils.py -------------------------------------------------------------------------------- /dataset_scripts/generate_multinli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/dataset_scripts/generate_multinli.py -------------------------------------------------------------------------------- /dataset_scripts/generate_waterbirds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/dataset_scripts/generate_waterbirds.py -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/loss.py -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/models.py -------------------------------------------------------------------------------- /run_expt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/run_expt.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/utils.py -------------------------------------------------------------------------------- /utils_glue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kohpangwei/group_DRO/HEAD/utils_glue.py --------------------------------------------------------------------------------