├── .gitignore ├── GOOD ├── __init__.py ├── data │ ├── __init__.py │ ├── dataset_manager.py │ ├── good_datasets │ │ ├── __init__.py │ │ ├── fiif_piif_motif.py │ │ ├── good_cmnist.py │ │ ├── good_hiv.py │ │ ├── good_motif.py │ │ ├── good_motif2.py │ │ ├── good_sst2.py │ │ ├── good_twitter.py │ │ └── lbap_core.py │ └── good_loaders │ │ ├── BaseLoader.py │ │ └── __init__.py ├── definitions.py ├── kernel │ ├── __init__.py │ ├── launch.py │ ├── launcher_manager.py │ ├── launchers │ │ ├── __init__.py │ │ ├── basic_launcher.py │ │ ├── harvest_launcher.py │ │ ├── multi_launcher.py │ │ └── single_launcher.py │ ├── main.py │ ├── pipeline_manager.py │ └── pipelines │ │ ├── __init__.py │ │ └── basic_pipeline.py ├── networks │ ├── __init__.py │ ├── model_manager.py │ └── models │ │ ├── ASAPGNNs.py │ │ ├── BaseGNN.py │ │ ├── CIGAGNNs.py │ │ ├── Classifiers.py │ │ ├── CoralNN.py │ │ ├── DANNs.py │ │ ├── DIRGNN.py │ │ ├── GINs.py │ │ ├── GINvirtualnode.py │ │ ├── GSATGNNs.py │ │ ├── LECIGNN.py │ │ ├── MolEncoders.py │ │ ├── Pooling.py │ │ └── __init__.py ├── ood_algorithms │ ├── __init__.py │ ├── algorithms │ │ ├── ASAP.py │ │ ├── BaseOOD.py │ │ ├── CIGA.py │ │ ├── Coral.py │ │ ├── DANN.py │ │ ├── DIR.py │ │ ├── ERM.py │ │ ├── GSAT.py │ │ ├── IRM.py │ │ ├── LECI.py │ │ ├── VREx.py │ │ └── __init__.py │ └── ood_manager.py └── utils │ ├── __init__.py │ ├── args.py │ ├── config_reader.py │ ├── data.py │ ├── evaluation.py │ ├── fast_pytorch_kmeans │ ├── __init__.py │ ├── init_methods.py │ ├── kmeans.py │ └── multi_kmeans.py │ ├── graph_visualize.py │ ├── initial.py │ ├── logger.py │ ├── metric.py │ ├── register.py │ ├── synthetic_data │ ├── BA3_loc.py │ ├── __init__.py │ └── synthetic_structsim.py │ └── train.py ├── LICENSE ├── README.md ├── configs ├── GOOD_configs │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ ├── ASAP.yaml │ │ │ ├── CIGA.yaml │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── IRM.yaml │ │ │ ├── LECI.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ ├── ASAP.yaml │ │ │ ├── CIGA.yaml │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── IRM.yaml │ │ │ ├── LECI.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ ├── ASAP.yaml │ │ │ ├── CIGA.yaml │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── IRM.yaml │ │ │ ├── LECI.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ ├── CIGA.yaml │ │ │ │ ├── DIR.yaml │ │ │ │ ├── ERM.yaml │ │ │ │ ├── GSAT.yaml │ │ │ │ ├── LECI.yaml │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ ├── CIGA.yaml │ │ │ │ ├── DIR.yaml │ │ │ │ ├── ERM.yaml │ │ │ │ ├── GSAT.yaml │ │ │ │ ├── LECI.yaml │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── ASAP.yaml │ │ │ ├── CIGA.yaml │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── IRM.yaml │ │ │ ├── LECI.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ └── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ ├── CIGA.yaml │ │ │ │ ├── DIR.yaml │ │ │ │ ├── ERM.yaml │ │ │ │ ├── GSAT.yaml │ │ │ │ ├── LECI.yaml │ │ │ │ └── base.yaml │ │ └── base.yaml │ └── base.yaml ├── final_configs │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ ├── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ │ │ └── no_shift │ │ │ ├── CIGA.yaml │ │ │ ├── CIGAv2.yaml │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── GroupDRO.yaml │ │ │ ├── IRM.yaml │ │ │ ├── Mixup.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ ├── covariate │ │ │ │ ├── CIGA.yaml │ │ │ │ ├── DIR.yaml │ │ │ │ ├── ERM.yaml │ │ │ │ ├── GSAT.yaml │ │ │ │ ├── LECI.yaml │ │ │ │ └── base.yaml │ │ │ └── no_shift │ │ │ │ ├── CIGA.yaml │ │ │ │ ├── CIGAv2.yaml │ │ │ │ ├── Coral.yaml │ │ │ │ ├── DANN.yaml │ │ │ │ ├── DIR.yaml │ │ │ │ ├── ERM.yaml │ │ │ │ ├── GSAT.yaml │ │ │ │ ├── GroupDRO.yaml │ │ │ │ ├── IRM.yaml │ │ │ │ ├── Mixup.yaml │ │ │ │ ├── VREx.yaml │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ ├── CIGA.yaml │ │ │ │ ├── DIR.yaml │ │ │ │ ├── ERM.yaml │ │ │ │ ├── GSAT.yaml │ │ │ │ ├── LECI.yaml │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ ├── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ │ │ └── no_shift │ │ │ ├── CIGA.yaml │ │ │ ├── CIGAv2.yaml │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── GroupDRO.yaml │ │ │ ├── IRM.yaml │ │ │ ├── Mixup.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ └── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ ├── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ │ │ └── no_shift │ │ │ ├── CIGA.yaml │ │ │ ├── CIGAv2.yaml │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── GroupDRO.yaml │ │ │ ├── IRM.yaml │ │ │ ├── Mixup.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ ├── covariate │ │ │ ├── CIGA.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── LECI.yaml │ │ │ └── base.yaml │ │ │ └── no_shift │ │ │ ├── Coral.yaml │ │ │ ├── DANN.yaml │ │ │ ├── DIR.yaml │ │ │ ├── ERM.yaml │ │ │ ├── GSAT.yaml │ │ │ ├── GroupDRO.yaml │ │ │ ├── IRM.yaml │ │ │ ├── Mixup.yaml │ │ │ ├── VREx.yaml │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ ├── CIGA.yaml │ │ │ │ ├── DIR.yaml │ │ │ │ ├── ERM.yaml │ │ │ │ ├── GSAT.yaml │ │ │ │ ├── LECI.yaml │ │ │ │ └── base.yaml │ │ └── base.yaml │ └── base.yaml └── sweep_configs │ ├── ASAP │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── CIGA │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── Coral │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── DANN │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── DIR │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── ERM │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── FPIIFMotif.yaml │ ├── GOODCMNIST.yaml │ ├── GOODHIV.yaml │ ├── GOODMotif.yaml │ ├── GOODMotif2.yaml │ ├── GOODSST2.yaml │ ├── GOODTwitter.yaml │ ├── GSAT │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── IRM │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ ├── LBAPcore.yaml │ ├── LECI │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ └── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── base.yaml │ └── base.yaml │ ├── VREx │ ├── FPIIFMotif │ │ ├── base.yaml │ │ └── basis │ │ │ ├── FIIF │ │ │ └── base.yaml │ │ │ ├── PIIF │ │ │ └── base.yaml │ │ │ ├── base.yaml │ │ │ ├── concept │ │ │ └── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODCMNIST │ │ ├── base.yaml │ │ └── color │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODHIV │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODMotif2 │ │ ├── base.yaml │ │ ├── basis │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODSST2 │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── GOODTwitter │ │ ├── base.yaml │ │ └── length │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ ├── LBAPcore │ │ ├── assay │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ ├── base.yaml │ │ ├── scaffold │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ │ └── base.yaml │ │ └── size │ │ │ ├── base.yaml │ │ │ └── covariate │ │ │ └── base.yaml │ └── base.yaml │ └── base.yaml ├── environment.yml └── setup.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/** 2 | storage 3 | docs/build 4 | /docs/source/_autosummary/ 5 | **/__pycace__/ 6 | /GOOD/kernel/auto_launch.py 7 | /GOOD/kernel/auto_chart.py 8 | /GOOD/kernel/auto_chart_final.py 9 | /GOOD/kernel/auto_chart_paper.py 10 | /GOOD/kernel/auto_chart_tvt.py 11 | /GOOD/kernel/auto_checkpoint.py 12 | /GOOD/kernel/auto_curve.py 13 | /GOOD/kernel/auto_curve_chart.py 14 | /configs/auto_config_finetune/ 15 | /graphEx/ 16 | /debug_log.py 17 | /GOOD/kernel/launchers/ada_launcher.py 18 | -------------------------------------------------------------------------------- /GOOD/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import config_summoner, args_parser 2 | from .utils.register import register 3 | from . import data, networks, ood_algorithms -------------------------------------------------------------------------------- /GOOD/data/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This data module includes 11 GOOD datasets and a dataloader for an organized data loading process. 3 | """ 4 | from GOOD.data.dataset_manager import load_dataset, create_dataloader 5 | from .good_datasets import * 6 | from .good_loaders import * 7 | -------------------------------------------------------------------------------- /GOOD/data/good_datasets/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module includes 11 GOOD datasets. 3 | 4 | - Graph prediction datasets: GOOD-HIV, GOOD-PCBA, GOOD-ZINC, GOOD-SST2, GOOD-CMNIST, GOOD-Motif. 5 | - Node prediction datasets: GOOD-Cora, GOOD-Arxiv, GOOD-Twitch, GOOD-WebKB, GOOD-CBAS. 6 | """ 7 | 8 | import glob 9 | from os.path import dirname, basename, isfile, join 10 | 11 | modules = glob.glob(join(dirname(__file__), "*.py")) 12 | __all__ = [basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')] 13 | 14 | from . import * 15 | 16 | -------------------------------------------------------------------------------- /GOOD/data/good_loaders/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module includes GOOD dataloaders. 3 | 4 | - Paired Balanced dataloader: PairDataLoader 5 | """ 6 | 7 | import glob 8 | from os.path import dirname, basename, isfile, join 9 | 10 | modules = glob.glob(join(dirname(__file__), "*.py")) 11 | __all__ = [basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')] 12 | 13 | from . import * -------------------------------------------------------------------------------- /GOOD/definitions.py: -------------------------------------------------------------------------------- 1 | r""" 2 | Only for project usage. This module includes a preset project root and a storage root. 3 | """ 4 | import os 5 | 6 | # root outside the GOOD 7 | STORAGE_DIR = os.path.join(os.path.abspath(os.path.dirname(os.path.dirname(__file__))), 'storage') #: :str - Storage root=/storage/. 8 | ROOT_DIR = os.path.abspath(os.path.dirname(os.path.dirname(__file__))) #: str - Project root or Repo root. 9 | OOM_CODE = 88 10 | -------------------------------------------------------------------------------- /GOOD/kernel/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module contains project kernel configuring, training, testing, and evaluation pipelines. 3 | """ 4 | from .pipelines import * -------------------------------------------------------------------------------- /GOOD/kernel/launchers/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module includes Benchmarking launchers. 3 | """ 4 | 5 | import glob 6 | from os.path import dirname, basename, isfile, join 7 | 8 | modules = glob.glob(join(dirname(__file__), "*.py")) 9 | __all__ = [basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')] 10 | 11 | from . import * -------------------------------------------------------------------------------- /GOOD/kernel/launchers/single_launcher.py: -------------------------------------------------------------------------------- 1 | import shlex 2 | import subprocess 3 | 4 | from tqdm import tqdm 5 | 6 | from GOOD import register 7 | from .basic_launcher import Launcher 8 | 9 | 10 | @register.launcher_register 11 | class SingleLauncher(Launcher): 12 | def __init__(self): 13 | super(SingleLauncher, self).__init__() 14 | 15 | def __call__(self, jobs_group, auto_args): 16 | jobs_group = super(SingleLauncher, self).__call__(jobs_group, auto_args) 17 | for cmd_args in tqdm(jobs_group): 18 | subprocess.run(shlex.split(cmd_args) + ['--gpu_idx', f'{auto_args.allow_devices[0]}'], close_fds=True, 19 | stdout=open('/dev/null', 'a'), stderr=open('/dev/null', 'a'), 20 | start_new_session=False) 21 | -------------------------------------------------------------------------------- /GOOD/kernel/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module includes Benchmarking pipelines. 3 | """ 4 | 5 | import glob 6 | from os.path import dirname, basename, isfile, join 7 | 8 | modules = glob.glob(join(dirname(__file__), "*.py")) 9 | __all__ = [basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')] 10 | 11 | from . import * -------------------------------------------------------------------------------- /GOOD/networks/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | A module that design GNN structures and manage GNN loadings. 3 | """ 4 | from .model_manager import load_model 5 | from .models import * 6 | -------------------------------------------------------------------------------- /GOOD/networks/model_manager.py: -------------------------------------------------------------------------------- 1 | r"""A module that is consist of a GNN model loader and model configuration function. 2 | """ 3 | 4 | import torch 5 | 6 | from GOOD import register 7 | from GOOD.utils.config_reader import Union, CommonArgs, Munch 8 | from GOOD.utils.initial import reset_random_seed 9 | 10 | 11 | def load_model(name: str, config: Union[CommonArgs, Munch]) -> torch.nn.Module: 12 | r""" 13 | A model loader. 14 | Args: 15 | name (str): Name of the chosen GNN. 16 | config (Union[CommonArgs, Munch]): Please refer to specific GNNs for required configs and formats. 17 | 18 | Returns: 19 | A instantiated GNN model. 20 | 21 | """ 22 | try: 23 | reset_random_seed(config) 24 | model = register.models[name](config) 25 | except KeyError as e: 26 | print(f'#E#Model {name} dose not exist.') 27 | raise e 28 | return model 29 | 30 | 31 | 32 | -------------------------------------------------------------------------------- /GOOD/networks/models/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module includes GNNs used in our leaderboard. It includes: GINs, GINvirtualnodes, and GCNs, in which GCNs are only 3 | for node classifications. 4 | """ 5 | 6 | import glob 7 | from os.path import dirname, basename, isfile, join 8 | 9 | modules = glob.glob(join(dirname(__file__), "*.py")) 10 | __all__ = [basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')] 11 | 12 | from . import * 13 | -------------------------------------------------------------------------------- /GOOD/ood_algorithms/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module is comprised by 7 OOD algorithms and their loader. 3 | """ 4 | 5 | from .algorithms import * 6 | -------------------------------------------------------------------------------- /GOOD/ood_algorithms/algorithms/ASAP.py: -------------------------------------------------------------------------------- 1 | """ 2 | Implementation of the baseline ERM 3 | """ 4 | from GOOD import register 5 | from GOOD.utils.config_reader import Union, CommonArgs, Munch 6 | from .BaseOOD import BaseOODAlg 7 | 8 | 9 | @register.ood_alg_register 10 | class ASAP(BaseOODAlg): 11 | r""" 12 | Implementation of the baseline ERM 13 | 14 | Args: 15 | config (Union[CommonArgs, Munch]): munchified dictionary of args 16 | """ 17 | def __init__(self, config: Union[CommonArgs, Munch]): 18 | super(ASAP, self).__init__(config) 19 | -------------------------------------------------------------------------------- /GOOD/ood_algorithms/algorithms/ERM.py: -------------------------------------------------------------------------------- 1 | """ 2 | Implementation of the baseline ERM 3 | """ 4 | from GOOD import register 5 | from GOOD.utils.config_reader import Union, CommonArgs, Munch 6 | from .BaseOOD import BaseOODAlg 7 | 8 | 9 | @register.ood_alg_register 10 | class ERM(BaseOODAlg): 11 | r""" 12 | Implementation of the baseline ERM 13 | 14 | Args: 15 | config (Union[CommonArgs, Munch]): munchified dictionary of args 16 | """ 17 | def __init__(self, config: Union[CommonArgs, Munch]): 18 | super(ERM, self).__init__(config) 19 | -------------------------------------------------------------------------------- /GOOD/ood_algorithms/algorithms/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module is for OOD algorithms. It is a good idea to inherit the BaseOOD class for new OOD algorithm designs. 3 | """ 4 | 5 | import glob 6 | from os.path import dirname, basename, isfile, join 7 | 8 | modules = glob.glob(join(dirname(__file__), "*.py")) 9 | __all__ = [basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')] 10 | 11 | from . import * 12 | -------------------------------------------------------------------------------- /GOOD/ood_algorithms/ood_manager.py: -------------------------------------------------------------------------------- 1 | """A module that is consist of an OOD algorithm loader. 2 | """ 3 | 4 | from GOOD import register 5 | from GOOD.ood_algorithms.algorithms.BaseOOD import BaseOODAlg 6 | from GOOD.utils.config_reader import Union, CommonArgs, Munch 7 | 8 | 9 | def load_ood_alg(name, config: Union[CommonArgs, Munch]): 10 | r""" 11 | OOD algorithm loader. 12 | Args: 13 | name: Name of the chosen OOD algorithm. 14 | config: please refer to specific algorithms for required configs. 15 | 16 | Returns: 17 | An OOD algorithm object. 18 | 19 | """ 20 | try: 21 | ood_algorithm: BaseOODAlg = register.ood_algs[name](config) 22 | except KeyError as e: 23 | print(f'#E#OOD algorithm of given name does not exist.') 24 | raise e 25 | return ood_algorithm 26 | -------------------------------------------------------------------------------- /GOOD/utils/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | This module contains various utilities, such as a component register, an argument parser, a configuration reader, and 3 | a metric object. 4 | """ 5 | from .config_reader import config_summoner 6 | from .args import args_parser 7 | from .logger import load_logger 8 | -------------------------------------------------------------------------------- /GOOD/utils/fast_pytorch_kmeans/__init__.py: -------------------------------------------------------------------------------- 1 | from .kmeans import KMeans 2 | from .multi_kmeans import MultiKMeans -------------------------------------------------------------------------------- /GOOD/utils/initial.py: -------------------------------------------------------------------------------- 1 | r"""Initial process for fixing all possible random seed. 2 | """ 3 | 4 | import random 5 | 6 | import numpy as np 7 | import torch 8 | 9 | from GOOD.utils.config_reader import Union, CommonArgs, Munch 10 | 11 | 12 | def reset_random_seed(config: Union[CommonArgs, Munch]): 13 | r""" 14 | Initial process for fixing all possible random seed. 15 | 16 | Args: 17 | config (Union[CommonArgs, Munch]): munchified dictionary of args (:obj:`config.random_seed`) 18 | 19 | 20 | """ 21 | # Fix Random seed 22 | random.seed(config.random_seed) 23 | np.random.seed(config.random_seed) 24 | torch.manual_seed(config.random_seed) 25 | torch.cuda.manual_seed(config.random_seed) 26 | torch.cuda.manual_seed_all(config.random_seed) 27 | torch.backends.cudnn.deterministic = True 28 | torch.backends.cudnn.benchmark = False 29 | 30 | # Default state is a training state 31 | torch.enable_grad() 32 | -------------------------------------------------------------------------------- /GOOD/utils/synthetic_data/__init__.py: -------------------------------------------------------------------------------- 1 | r""" 2 | Utilities for generating certain graph shapes. 3 | 4 | Copied from `gnn-model-explainer `_ 5 | """ -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 8 | pre_train: 100 -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/ASAP.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: ASAPvGIN 5 | ood: 6 | ood_alg: ASAP 7 | ood_param: 0.25 # causal ratio 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | 14 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 # [Subgraph discovery architecture hyperparameter] GSAT IB constraint for stable training. Enable/Disable: 1.0/0.0 8 | extra_param: 9 | - 0.1 # 5 # LA: label adversarial 10 | - 0 # Never used 11 | - 0.3 # 20 # EA: environment adversarial 12 | - 50 # [Subgraph discovery architecture hyperparameter] GSAT subgraph searching decay interval control (only used when ood_param = 1.0) 13 | - 0 # Pure feature shift consideration: the input pre-transform hyperparameter. 14 | - true # Never used 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 1000. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: FIIF -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/ASAP.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: ASAPvGIN 5 | ood: 6 | ood_alg: ASAP 7 | ood_param: 0.25 # causal ratio 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | 14 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 # [Subgraph discovery architecture hyperparameter] GSAT IB constraint for stable training. Enable/Disable: 1.0/0.0 8 | extra_param: 9 | - 5 # LA: label adversarial 10 | - 0 # Never used 11 | - 20 # EA: environment adversarial 12 | - 50 # [Subgraph discovery architecture hyperparameter] GSAT subgraph searching decay interval control (only used when ood_param = 1.0) 13 | - 0 # Pure feature shift consideration: the input pre-transform hyperparameter. 14 | - true # Never used 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0001 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 1000. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: PIIF -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: FPIIFMotif 5 | domain: basis 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/concept/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/concept/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/concept/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/concept/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/concept/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 # [Subgraph discovery architecture hyperparameter] GSAT IB constraint for stable training. Enable/Disable: 1.0/0.0 8 | extra_param: 9 | - 5 # LA: label adversarial 10 | - 0 # Never used 11 | - 20 # EA: environment adversarial 12 | - 50 # [Subgraph discovery architecture hyperparameter] GSAT subgraph searching decay interval control (only used when ood_param = 1.0) 13 | - 0 # Pure feature shift consideration: the input pre-transform hyperparameter. 14 | - true # Never used 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: FIIF -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/ASAP.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: ASAPvGIN 5 | ood: 6 | ood_alg: ASAP 7 | ood_param: 0.25 # causal ratio 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | 14 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 # [Subgraph discovery architecture hyperparameter] GSAT IB constraint for stable training. Enable/Disable: 1.0/0.0 8 | extra_param: 9 | - 0.1 # 5 # LA: label adversarial 10 | - 0 # Never used 11 | - 0.3 # 20 # EA: environment adversarial 12 | - 50 # [Subgraph discovery architecture hyperparameter] GSAT subgraph searching decay interval control (only used when ood_param = 1.0) 13 | - 0 # Pure feature shift consideration: the input pre-transform hyperparameter. 14 | - true # Never used 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 1000. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODCMNIST 5 | model: 6 | model_layer: 5 7 | global_pool: mean 8 | train: 9 | num_steps: 10 -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: color 5 | train: 6 | max_epoch: 500 7 | train_bs: 128 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/color/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 4 10 | - 1 11 | train: 12 | max_epoch: 100 13 | lr: 0.001 14 | mile_stones: 15 | - 300 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/color/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.6 8 | extra_param: 9 | - 0.01 10 | train: 11 | max_epoch: 100 12 | lr: 0.001 13 | mile_stones: 14 | - 300 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/color/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 500 10 | lr: 0.001 11 | mile_stones: 12 | - 300 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/color/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 10 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 300 17 | train_bs: 64 18 | val_bs: 128 19 | test_bs: 128 20 | weight_decay: 0.0 21 | log_file: exp_0 22 | clean_save: true 23 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/color/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 100 10 | - 0 11 | - 1 12 | - 10 13 | - 1 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | pre_train: 50 19 | mile_stones: 20 | - 300 21 | train_bs: 64 22 | val_bs: 128 23 | test_bs: 128 24 | weight_decay: 0.0 25 | log_file: exp_0 26 | clean_save: true 27 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 8 | mile_stones: [150] -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODHIV 5 | domain: scaffold 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/scaffold/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 1 10 | - 2 11 | train: 12 | max_epoch: 200 13 | train_bs: 32 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/scaffold/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.1 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/scaffold/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 300 10 | lr: 0.0001 11 | weight_decay: 0.0001 12 | log_file: exp_0 13 | clean_save: true 14 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/scaffold/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.5 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/scaffold/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 1 10 | - 0 11 | - 0.01 12 | - 60 13 | - 0 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 50 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODHIV 5 | domain: size 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/size/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.2 8 | extra_param: 9 | - 1 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | train_bs: 32 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/size/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.1 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/size/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 300 10 | train_bs: 64 11 | lr: 0.0001 12 | weight_decay: 0.0001 13 | log_file: exp_0 14 | clean_save: true 15 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/size/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.5 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/size/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 0.1 10 | - 0 11 | - 0.1 12 | - 60 13 | - 0.1 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 50 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODMotif 5 | domain: basis 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/basis/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/basis/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/basis/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/basis/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/basis/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 # [Subgraph discovery architecture hyperparameter] GSAT IB constraint for stable training. Enable/Disable: 1.0/0.0 8 | extra_param: 9 | - 5 # LA: label adversarial 10 | - 0 # Never used 11 | - 20 # EA: environment adversarial 12 | - 50 # [Subgraph discovery architecture hyperparameter] GSAT subgraph searching decay interval control (only used when ood_param = 1.0) 13 | - 0 # Pure feature shift consideration: the input pre-transform hyperparameter. 14 | - true # Never used 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODMotif 5 | domain: size 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/ASAP.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: ASAPvGIN 5 | ood: 6 | ood_alg: ASAP 7 | ood_param: 0.25 # causal ratio 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | 14 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.01 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 1 10 | - 0 11 | - 0.3 12 | - 50 13 | - 0 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 200 20 | pre_train: 50 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 1000. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | dim_hidden: 300 7 | dim_ffn: 300 8 | dropout_rate: 0.5 9 | train: 10 | num_steps: 10 11 | max_epoch: 200 12 | train_bs: 32 13 | val_bs: 256 14 | test_bs: 256 15 | weight_decay: 0. 16 | dataset: 17 | dataset_name: GOODMotif2 18 | domain: basis 19 | generate: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: basis 5 | train: 6 | max_epoch: 200 7 | train_bs: 32 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/basis/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.5 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/basis/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.01 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/basis/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/basis/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 10 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/basis/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 10 10 | - 0 11 | - 10 12 | - 20 13 | - 0 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 8 | mile_stones: [150] -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODSST2 5 | domain: length 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/length/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | weight_decay: 0.0 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/length/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/length/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | weight_decay: 0.0 12 | log_file: exp_0 13 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/length/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/length/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 0.1 10 | - 0 11 | - 0.1 12 | - 60 13 | - 0.01 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | pre_train: 50 19 | weight_decay: 0.0 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODTwitter 5 | model: 6 | model_layer: 3 7 | global_pool: mean 8 | train: 9 | num_steps: 10 10 | mile_stones: [150] -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: length 5 | train: 6 | max_epoch: 200 7 | train_bs: 32 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/length/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 1 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | pre_train: 5 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/length/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.6 8 | extra_param: 9 | - 1 10 | train: 11 | max_epoch: 100 12 | lr: 0.0001 13 | weight_decay: 0.0001 14 | pre_train: 5 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/length/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.0001 11 | weight_decay: 0.0001 12 | pre_train: 5 13 | log_file: exp_0 14 | clean_save: true 15 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/length/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 10 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | pre_train: 5 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/length/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 0.1 10 | - 0 11 | - 0.001 12 | - 30 13 | - 0.001 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 5 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/GOOD_configs/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: assay 5 | train: 6 | max_epoch: 200 7 | train_bs: 32 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/assay/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 4 10 | - 1 11 | train: 12 | max_epoch: 200 13 | train_bs: 32 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/assay/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 1 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/assay/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 300 10 | lr: 0.0001 11 | weight_decay: 0.0001 12 | log_file: exp_0 13 | clean_save: true 14 | -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/assay/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.5 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/assay/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 1 10 | - 0 11 | - 0.01 12 | - 90 13 | - 0.1 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 20 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/GOOD_configs/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 8 | mile_stones: [150] 9 | dataset: 10 | dataset_name: LBAPcore -------------------------------------------------------------------------------- /configs/GOOD_configs/base.yaml: -------------------------------------------------------------------------------- 1 | task: train 2 | random_seed: 123 3 | exp_round: null 4 | log_file: default 5 | gpu_idx: 0 6 | ckpt_root: null 7 | ckpt_dir: null 8 | save_tag: null 9 | other_saved: null 10 | clean_save: False 11 | pytest: False 12 | pipeline: Pipeline 13 | num_workers: 0 14 | train: 15 | weight_decay: 0. 16 | save_gap: 10 17 | tr_ctn: False 18 | epoch: 0 19 | ctn_epoch: 0 20 | alpha: 0.2 # parameter for DANN 21 | stage_stones: [] 22 | pre_train: 0 23 | model: 24 | dim_hidden: 300 25 | dim_ffn: 300 26 | dropout_rate: 0.5 27 | dataset: 28 | dataloader_name: BaseDataLoader 29 | dataset_root: null 30 | generate: False 31 | dim_node: null 32 | dim_edge: null 33 | num_classes: null 34 | num_envs: null 35 | feat_dims: null 36 | edge_feat_dims: null 37 | ood: 38 | extra_param: null -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODCMNIST 5 | model: 6 | model_layer: 5 7 | global_pool: mean 8 | train: 9 | num_steps: 10 -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: color 5 | train: 6 | max_epoch: 500 7 | train_bs: 128 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 4 10 | - 1 11 | train: 12 | max_epoch: 100 13 | lr: 0.001 14 | mile_stones: 15 | - 300 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.6 8 | extra_param: 9 | - 0.01 10 | train: 11 | max_epoch: 100 12 | lr: 0.001 13 | mile_stones: 14 | - 300 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 500 10 | lr: 0.001 11 | mile_stones: 12 | - 300 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 10 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 300 17 | train_bs: 64 18 | val_bs: 128 19 | test_bs: 128 20 | weight_decay: 0.0 21 | log_file: exp_0 22 | clean_save: true 23 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 100 10 | - 0 11 | - 1 12 | - 10 13 | - 1 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | pre_train: 50 19 | mile_stones: 20 | - 300 21 | train_bs: 64 22 | val_bs: 128 23 | test_bs: 128 24 | weight_decay: 0.0 25 | log_file: exp_0 26 | clean_save: true 27 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 # causal ratio 8 | extra_param: 9 | - 1 10 | train: 11 | max_epoch: 100 12 | lr: 1e-3 13 | mile_stones: [300] 14 | 15 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/CIGAv2.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 # causal ratio 8 | extra_param: 9 | - 1 10 | - 1 11 | train: 12 | max_epoch: 100 13 | lr: 1e-3 14 | mile_stones: [300] 15 | 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 0.001 8 | train: 9 | max_epoch: 500 10 | lr: 1e-3 11 | mile_stones: [300] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.001 8 | train: 9 | max_epoch: 500 10 | lr: 1e-3 11 | mile_stones: [300] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.6 # causal ratio 8 | extra_param: 9 | - 1e-4 10 | train: 11 | max_epoch: 100 12 | lr: 1e-3 13 | mile_stones: [300] 14 | 15 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1. 8 | train: 9 | max_epoch: 500 10 | lr: 1e-3 11 | mile_stones: [300] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 # info loss coefficient 8 | extra_param: 9 | - false # learn_edge_att 10 | - 20 # decay_interval 11 | - 0.5 # final_r 12 | train: 13 | max_epoch: 200 14 | lr: 1e-3 15 | mile_stones: [300] 16 | 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/GroupDRO.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: GroupDRO 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 500 10 | lr: 1e-3 11 | mile_stones: [300] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 1. 8 | train: 9 | max_epoch: 500 10 | lr: 1e-3 11 | mile_stones: [300] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/Mixup.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Mixup_vGIN 5 | ood: 6 | ood_alg: Mixup 7 | ood_param: 0.4 8 | train: 9 | max_epoch: 500 10 | lr: 1e-3 11 | mile_stones: [300] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 1e-1 8 | train: 9 | max_epoch: 500 10 | lr: 1e-3 11 | mile_stones: [300] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODCMNIST/color/no_shift/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: no_shift -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 8 | mile_stones: [150] -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODHIV 5 | domain: scaffold 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 1 10 | - 2 11 | train: 12 | max_epoch: 200 13 | train_bs: 32 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.1 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 300 10 | lr: 0.0001 11 | weight_decay: 0.0001 12 | log_file: exp_0 13 | clean_save: true 14 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.5 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 0.1 10 | - 0 11 | - 0.01 12 | - 60 13 | - 0.1 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 50 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 # causal ratio 8 | extra_param: 9 | - 1 # additional loss control 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 1e-3 14 | 15 | 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/CIGAv2.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 # causal ratio 8 | extra_param: 9 | - 1 # additional loss control 10 | - 1 11 | train: 12 | max_epoch: 200 13 | train_bs: 32 14 | lr: 1e-3 15 | 16 | 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 # causal ratio 8 | extra_param: 9 | - 1e-4 # additional loss control 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 1e-3 14 | 15 | 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1. 8 | train: 9 | max_epoch: 250 10 | lr: 1e-3 11 | mile_stones: [190] 12 | 13 | 14 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 # info loss coefficient 8 | extra_param: 9 | - true # learn_edge_att 10 | - 20 # decay_interval 11 | - 0.5 # final_r 12 | train: 13 | max_epoch: 200 14 | lr: 1e-3 15 | 16 | 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/GroupDRO.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: GroupDRO 7 | ood_param: 0.001 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/Mixup.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Mixup_vGIN 5 | ood: 6 | ood_alg: Mixup 7 | ood_param: 2. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 100. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/scaffold/no_shift/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: no_shift 5 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODHIV 5 | domain: size 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/size/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.2 8 | extra_param: 9 | - 1 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | train_bs: 32 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/size/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.1 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/size/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 300 10 | train_bs: 64 11 | lr: 0.0001 12 | weight_decay: 0.0001 13 | log_file: exp_0 14 | clean_save: true 15 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/size/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.5 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/size/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 1 10 | - 0 11 | - 0.1 12 | - 60 13 | - 0.1 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 50 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/final_configs/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODMotif 5 | domain: basis 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/basis/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 100 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/basis/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 100 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/basis/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/basis/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/basis/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 # [Subgraph discovery architecture hyperparameter] GSAT IB constraint for stable training. Enable/Disable: 1.0/0.0 8 | extra_param: 9 | - 10 # LA: label adversarial 10 | - 0 # Never used 11 | - 10 # EA: environment adversarial 12 | - 50 # [Subgraph discovery architecture hyperparameter] GSAT subgraph searching decay interval control (only used when ood_param = 1.0) 13 | - 0 # Pure feature shift consideration: the input pre-transform hyperparameter. 14 | - true # Never used 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODMotif 5 | domain: size 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 100 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.01 10 | train: 11 | max_epoch: 100 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 0.1 10 | - 0 11 | - 0.3 12 | - 50 13 | - 0 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 200 20 | pre_train: 50 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.25 # causal ratio 8 | extra_param: 9 | - 1 # additional loss control 10 | train: 11 | max_epoch: 200 12 | lr: 1e-3 13 | mile_stones: [150] 14 | 15 | 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/CIGAv2.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.25 # causal ratio 8 | extra_param: 9 | - 1 # additional loss control 10 | - 1 11 | train: 12 | max_epoch: 200 13 | lr: 1e-3 14 | mile_stones: [150] 15 | 16 | 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_GIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_GIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.25 # causal ratio 8 | extra_param: 9 | - 1e-2 # additional loss control 10 | train: 11 | max_epoch: 200 12 | lr: 1e-3 13 | mile_stones: [150] 14 | 15 | 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 # info loss coefficient 8 | extra_param: 9 | - true # learn_edge_att 10 | - 20 # decay_interval 11 | - 0.5 # final_r 12 | train: 13 | max_epoch: 200 14 | lr: 1e-3 15 | mile_stones: [150] 16 | 17 | 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/GroupDRO.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: GroupDRO 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/Mixup.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Mixup_GIN 5 | ood: 6 | ood_alg: Mixup 7 | ood_param: 0.4 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 1000. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | mile_stones: [150] 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif/size/no_shift/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: no_shift -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | dim_hidden: 300 7 | dim_ffn: 300 8 | dropout_rate: 0.5 9 | train: 10 | num_steps: 10 11 | max_epoch: 200 12 | train_bs: 32 13 | val_bs: 256 14 | test_bs: 256 15 | weight_decay: 0. 16 | dataset: 17 | dataset_name: GOODMotif2 18 | domain: basis 19 | generate: true 20 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: basis 5 | train: 6 | max_epoch: 200 7 | train_bs: 32 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/basis/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.5 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | mile_stones: 15 | - 150 16 | weight_decay: 0.0 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/basis/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 0.01 10 | train: 11 | max_epoch: 100 12 | lr: 0.001 13 | mile_stones: 14 | - 150 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/basis/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | mile_stones: 12 | - 150 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/basis/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1 8 | extra_param: 9 | - true 10 | - 10 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | mile_stones: 16 | - 150 17 | weight_decay: 0.0 18 | log_file: exp_0 19 | clean_save: true 20 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/basis/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 10 10 | - 0 11 | - 20 12 | - 20 13 | - 0 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | mile_stones: 19 | - 150 20 | pre_train: 100 21 | weight_decay: 0.0 22 | log_file: exp_0 23 | clean_save: true 24 | -------------------------------------------------------------------------------- /configs/final_configs/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 8 | mile_stones: [150] -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODSST2 5 | domain: length 6 | train: 7 | max_epoch: 200 8 | train_bs: 32 9 | val_bs: 256 10 | test_bs: 256 11 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 8 | extra_param: 9 | - 2 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.001 14 | weight_decay: 0.0 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 10.0 10 | train: 11 | max_epoch: 200 12 | lr: 0.001 13 | weight_decay: 0.0 14 | log_file: exp_0 15 | clean_save: true 16 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.001 11 | weight_decay: 0.0 12 | log_file: exp_0 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.001 15 | weight_decay: 0.0 16 | log_file: exp_0 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 0.0 8 | extra_param: 9 | - 0.01 10 | - 0 11 | - 0.1 12 | - 60 13 | - 0.01 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.001 18 | pre_train: 50 19 | weight_decay: 0.0 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 # causal ratio 8 | extra_param: 9 | - 1 #1e-4 10 | train: 11 | max_epoch: 100 12 | lr: 1e-3 -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/CIGAv2.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.8 # causal ratio 8 | extra_param: 9 | - 1 #1e-4 10 | - 1 11 | train: 12 | max_epoch: 100 13 | lr: 1e-3 -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 # causal ratio 8 | extra_param: 9 | - 10 #1e-4 10 | train: 11 | max_epoch: 100 12 | lr: 1e-3 -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1. 8 | train: 9 | max_epoch: 250 10 | lr: 1e-3 11 | mile_stones: [200] 12 | 13 | 14 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 # info loss coefficient 8 | extra_param: 9 | - true # learn_edge_att 10 | - 20 # decay_interval 11 | - 0.5 # final_r 12 | train: 13 | max_epoch: 200 14 | lr: 1e-3 -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/GroupDRO.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: GroupDRO 7 | ood_param: 0.001 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/Mixup.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Mixup_vGIN 5 | ood: 6 | ood_alg: Mixup 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 10. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODSST2/length/no_shift/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: no_shift 5 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | dataset_name: GOODTwitter 5 | model: 6 | model_layer: 3 7 | global_pool: mean 8 | train: 9 | num_steps: 10 10 | mile_stones: [150] -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: length 5 | train: 6 | max_epoch: 200 7 | train_bs: 32 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 1 10 | - 0.5 11 | train: 12 | max_epoch: 200 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | pre_train: 5 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.6 8 | extra_param: 9 | - 1 10 | train: 11 | max_epoch: 100 12 | lr: 0.0001 13 | weight_decay: 0.0001 14 | pre_train: 5 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 200 10 | lr: 0.0001 11 | weight_decay: 0.0001 12 | pre_train: 5 13 | log_file: exp_0 14 | clean_save: true 15 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 10 11 | - 0.7 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | pre_train: 5 17 | log_file: exp_0 18 | clean_save: true 19 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 0.1 10 | - 0 11 | - 0.001 12 | - 30 13 | - 0.001 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 5 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/Coral.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Coral_vGIN 5 | ood: 6 | ood_alg: Coral 7 | ood_param: 0.1 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/DANN.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DANN_vGIN 5 | ood: 6 | ood_alg: DANN 7 | ood_param: 0.01 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 # causal ratio 8 | extra_param: 9 | - 10 #1e-4 10 | train: 11 | max_epoch: 100 12 | lr: 1e-3 -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1. 8 | train: 9 | max_epoch: 250 10 | lr: 1e-3 11 | mile_stones: [200] 12 | 13 | 14 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 # info loss coefficient 8 | extra_param: 9 | - true # learn_edge_att 10 | - 20 # decay_interval 11 | - 0.5 # final_r 12 | train: 13 | max_epoch: 200 14 | lr: 1e-3 -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/GroupDRO.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: GroupDRO 7 | ood_param: 0.001 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/IRM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: IRM 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/Mixup.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: Mixup_vGIN 5 | ood: 6 | ood_alg: Mixup 7 | ood_param: 1. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/VREx.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: VREx 7 | ood_param: 10. 8 | train: 9 | max_epoch: 200 10 | lr: 1e-3 11 | 12 | 13 | -------------------------------------------------------------------------------- /configs/final_configs/GOODTwitter/length/no_shift/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: no_shift 5 | -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | domain: assay 5 | train: 6 | max_epoch: 200 7 | train_bs: 32 8 | val_bs: 256 9 | test_bs: 256 10 | weight_decay: 0. -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/assay/covariate/CIGA.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: CIGAvGIN 5 | ood: 6 | ood_alg: CIGA 7 | ood_param: 0.6 8 | extra_param: 9 | - 4 10 | - 1 11 | train: 12 | max_epoch: 200 13 | train_bs: 32 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/assay/covariate/DIR.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: DIRvGIN 5 | ood: 6 | ood_alg: DIR 7 | ood_param: 0.8 8 | extra_param: 9 | - 1 10 | train: 11 | max_epoch: 200 12 | train_bs: 32 13 | lr: 0.0001 14 | weight_decay: 0.0001 15 | log_file: exp_0 16 | clean_save: true 17 | -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/assay/covariate/ERM.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: vGIN 5 | ood: 6 | ood_alg: ERM 7 | ood_param: -1.0 8 | train: 9 | max_epoch: 300 10 | lr: 0.0001 11 | weight_decay: 0.0001 12 | log_file: exp_0 13 | clean_save: true 14 | -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/assay/covariate/GSAT.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: GSATvGIN 5 | ood: 6 | ood_alg: GSAT 7 | ood_param: 1.0 8 | extra_param: 9 | - true 10 | - 20 11 | - 0.5 12 | train: 13 | max_epoch: 200 14 | lr: 0.0001 15 | weight_decay: 0.0001 16 | log_file: exp_0 17 | clean_save: true 18 | -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/assay/covariate/LECI.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - base.yaml 3 | model: 4 | model_name: LECIvGIN 5 | ood: 6 | ood_alg: LECI 7 | ood_param: 1.0 8 | extra_param: 9 | - 1 10 | - 0 11 | - 0.1 12 | - 90 13 | - 0.1 14 | - true 15 | train: 16 | max_epoch: 200 17 | lr: 0.0001 18 | pre_train: 20 19 | weight_decay: 0.0001 20 | log_file: exp_0 21 | clean_save: true 22 | -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | dataset: 4 | shift_type: covariate -------------------------------------------------------------------------------- /configs/final_configs/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | model: 4 | model_layer: 3 5 | global_pool: mean 6 | train: 7 | num_steps: 10 8 | mile_stones: [150] 9 | dataset: 10 | dataset_name: LBAPcore -------------------------------------------------------------------------------- /configs/final_configs/base.yaml: -------------------------------------------------------------------------------- 1 | task: train 2 | random_seed: 123 3 | exp_round: null 4 | log_file: default 5 | gpu_idx: 0 6 | ckpt_root: null 7 | ckpt_dir: null 8 | save_tag: null 9 | other_saved: null 10 | clean_save: False 11 | pytest: False 12 | pipeline: Pipeline 13 | num_workers: 0 14 | train: 15 | weight_decay: 0. 16 | save_gap: 10 17 | tr_ctn: False 18 | epoch: 0 19 | ctn_epoch: 0 20 | alpha: 0.2 # parameter for DANN 21 | stage_stones: [] 22 | pre_train: 0 23 | model: 24 | dim_hidden: 300 25 | dim_ffn: 300 26 | dropout_rate: 0.5 27 | dataset: 28 | dataloader_name: BaseDataLoader 29 | dataset_root: null 30 | generate: False 31 | dim_node: null 32 | dim_edge: null 33 | num_classes: null 34 | num_envs: null 35 | feat_dims: null 36 | edge_feat_dims: null 37 | ood: 38 | extra_param: null -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ASAP/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: 4 | - 0.2 5 | - 0.4 6 | - 0.6 7 | - 0.8 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/CIGA/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: # causal ratio 4 | - 0.2 5 | - 0.4 6 | - 0.6 7 | - 0.8 8 | extra_param: 9 | - 10 | - 0.5 11 | - 1 12 | - 2 13 | - 4 14 | - 15 | - 0.5 16 | - 1 17 | - 2 18 | - 4 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/Coral/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: 4 | - 1 5 | - 1e-1 6 | - 1e-2 7 | - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DANN/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: 4 | - 1 5 | - 1e-1 6 | - 1e-2 7 | - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/DIR/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: # causal ratio 4 | - 0.2 5 | - 0.4 6 | - 0.6 7 | - 0.8 8 | extra_param: 9 | - # additional loss control 10 | - 1e1 11 | - 1 12 | - 1e-1 13 | - 1e-2 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/ERM/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/FPIIFMotif.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | # - 1e-3 9 | - 1e-4 10 | weight_decay: 11 | - 0. 12 | - 1e-4 -------------------------------------------------------------------------------- /configs/sweep_configs/GOODCMNIST.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | - 1e-3 9 | weight_decay: 10 | - 0. -------------------------------------------------------------------------------- /configs/sweep_configs/GOODHIV.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | - 1e-4 9 | weight_decay: 10 | - 1e-4 -------------------------------------------------------------------------------- /configs/sweep_configs/GOODMotif.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | # - 4 # LECI has these 7 extra runs. 6 | # - 5 7 | # - 6 8 | # - 7 9 | # - 8 10 | # - 9 11 | # - 10 12 | log_file: 13 | - exp_0 14 | lr: 15 | - 1e-3 16 | weight_decay: 17 | - 0. -------------------------------------------------------------------------------- /configs/sweep_configs/GOODMotif2.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | - 1e-3 9 | weight_decay: 10 | - 0. -------------------------------------------------------------------------------- /configs/sweep_configs/GOODSST2.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | - 1e-3 9 | weight_decay: 10 | - 0. -------------------------------------------------------------------------------- /configs/sweep_configs/GOODTwitter.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | - 1e-4 9 | weight_decay: 10 | - 1e-4 11 | pre_train: 12 | - 5 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/GSAT/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | extra_param: 4 | - # learn_edge_att 5 | - true 6 | - # decay_interval 7 | - 10 8 | - 20 9 | - # final_r 10 | - 0.5 11 | - 0.7 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/IRM/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: 4 | - 100 5 | - 10 6 | - 1 7 | - 1e-1 -------------------------------------------------------------------------------- /configs/sweep_configs/LBAPcore.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | - 1e-4 9 | weight_decay: 10 | - 1e-4 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | pre_train: 5 | - 100 6 | mile_stones: 7 | - 201 8 | -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: # information constraint 4 | - 0 5 | # - 1 6 | extra_param: 7 | - # LA 8 | - 1 9 | - 0.3 10 | - 0.1 11 | - 3e-2 12 | - 1e-2 13 | # - 0 14 | - # EC 15 | - 0 16 | - # EA 17 | - 0.3 18 | - 0.1 19 | - 3e-2 20 | - 1e-2 21 | - # r decay interval 22 | - 50 23 | - # EF 24 | - 0 25 | - # If batchnorm/instancenorm (true/false) 26 | - true -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | extra_param: 4 | - # LA 5 | - 0.1 6 | - 1e-2 7 | - # EC 8 | - 0 9 | - # EA 10 | - 0.1 11 | - 1e-2 12 | - # r decay interval 13 | - 50 14 | - # EF 15 | - 0 16 | - # If batchnorm/instancenorm (true/false) 17 | - true 18 | # - false 19 | -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: # no information constraint 4 | - 0 5 | extra_param: 6 | - # LA 7 | - 10 8 | - 1 9 | - 0.1 10 | # - 0 11 | - # EC 12 | - 0 13 | - # EA 14 | - 1 15 | - 0.1 16 | - 1e-2 17 | - # r decay interval 18 | - 50 19 | - # EF 20 | - 0 21 | - # If batchnorm/instancenorm (true/false) 22 | - true -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml 4 | ood_param: # information constraint 5 | - 0 6 | extra_param: 7 | - # LA 8 | - 100 9 | - 10 10 | # - 1 11 | # - 0 12 | - # EC 13 | - 0 14 | - # EA 15 | - 10 16 | - 1 17 | # - 0.1 18 | # - 0 19 | - # r decay interval 20 | - 10 21 | - # EF 22 | - 1 23 | - 0.1 24 | # - 0 25 | - 26 | - true -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml 4 | ood_param: 5 | - 1 6 | extra_param: 7 | - # LA 8 | - 1 9 | - 0.1 10 | - 0.01 11 | - # EC 12 | - 0 13 | - # EA 14 | - 0.1 15 | - 0.01 16 | # - 0.001 17 | - # r decay interval 18 | - 60 19 | - # EF 20 | - 0 21 | - 0.1 22 | - 0.01 23 | # - 0.001 24 | - 25 | - true 26 | # - false 27 | pre_train: 28 | - 50 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml 4 | ood_param: # information constraint 5 | - 0 6 | # - 1 7 | extra_param: 8 | - # LA 9 | - 10 10 | - 5 11 | - 1 12 | # - 0 13 | - # EC 14 | - 0 15 | - # EA 16 | - 20 17 | - 10 18 | # - 0 19 | - # r decay interval 20 | - 50 21 | - # EF 22 | - 0 23 | - # If batchnorm/instancenorm (true/false) 24 | - true 25 | pre_train: 26 | - 100 27 | 28 | #ood_param: # information constraint 29 | # - 0 30 | #extra_param: 31 | # - # LA 32 | # - 10 33 | # - 5 34 | # - 1 35 | ## - 0 36 | # - # EC 37 | # - 0 38 | # - # EA 39 | # - 10 40 | # - 5 41 | # - 1 42 | # - # r decay interval 43 | # - 50 44 | # - # EF 45 | # - 0 46 | # - # If batchnorm/instancenorm (true/false) 47 | # - true 48 | #pre_train: 49 | # - 50 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: # information constraint 4 | - 0 5 | extra_param: 6 | - # LA 7 | - 1 8 | - 0.1 9 | # - 0 10 | - # EC 11 | - 0 12 | - # EA 13 | - 1 14 | - 0.3 15 | - 0.1 16 | - # r decay interval 17 | - 50 18 | - # EF 19 | - 0 20 | - # If batchnorm/instancenorm (true/false) 21 | - true 22 | pre_train: 23 | - 50 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | ood_param: # information constraint 9 | - 0 10 | extra_param: 11 | - # LA 12 | - 10 13 | - 5 14 | - 1 15 | # - 0 16 | - # EC 17 | - 0 18 | # - 0.001 19 | - # EA 20 | - 20 21 | - 10 22 | # - 0 23 | - # r decay interval 24 | - 20 25 | - # EF 26 | - 0 27 | - # If batchnorm/instancenorm (true/false) 28 | - true 29 | pre_train: 30 | - 100 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | ood_param: 5 | - 0 6 | - 1 7 | extra_param: 8 | - # LA 9 | - 1 10 | - 0.1 11 | - 0.01 12 | # - 0 13 | - # EC 14 | - 0 15 | - # EA 16 | - 0.1 17 | - 0.01 18 | # - 0 19 | - # r decay interval 20 | - 60 21 | - # EF 22 | # - 0 23 | - 0.1 24 | - 0.01 25 | - 26 | - true 27 | pre_train: 28 | - 50 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml 4 | ood_param: 5 | - 1 6 | extra_param: 7 | - # LA 8 | - 1 9 | - 0.1 10 | - 0.01 11 | # - 0 12 | - # EC 13 | - 0 14 | - # EA 15 | - 0.1 16 | - 0.01 17 | - 0.001 18 | # - 0 19 | - # r decay interval 20 | - 30 21 | - # EF 22 | # - 0 23 | - 0.1 24 | - 0.01 25 | - 0.001 26 | - 27 | - true 28 | pre_train: 29 | - 5 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml 4 | ood_param: # information constraint 5 | - 1 6 | extra_param: 7 | - # LA 8 | - 1 9 | - 0.1 10 | - 0.01 11 | - # EC 12 | - 0 13 | - # EA 14 | - 0.1 15 | - 0.01 16 | - 0.001 17 | - # r decay interval 18 | - 90 19 | - # EF 20 | - 0.1 21 | - 0.01 22 | - # If batchnorm/instancenorm (true/false) 23 | - true 24 | pre_train: 25 | - 20 -------------------------------------------------------------------------------- /configs/sweep_configs/LECI/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: 4 | - 1 5 | extra_param: 6 | - # LA 7 | - 1 8 | - 0.1 9 | - 0.01 10 | - # EC 11 | - 0 12 | - # EA 13 | - 0.1 14 | - 0.01 15 | # - 0.001 16 | - # r decay interval 17 | - 60 18 | - # EF 19 | - 0 20 | - 0.1 21 | - 0.01 22 | # - 0.001 23 | - 24 | - true 25 | pre_train: 26 | - 50 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/FPIIFMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../FPIIFMotif.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/FPIIFMotif/basis/FIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/FPIIFMotif/basis/PIIF/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/FPIIFMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/FPIIFMotif/basis/concept/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/FPIIFMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODCMNIST/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODCMNIST.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODCMNIST/color/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODCMNIST/color/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODHIV/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODHIV.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODHIV/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODHIV/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODHIV/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODHIV/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODMotif.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | lr: 4 | - 1e-3 5 | # - 1e-4 6 | weight_decay: 7 | - 0. 8 | # - 1e-3 9 | # - 1e-4 10 | # - 1e-5 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif2/basis/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif2/basis/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif2/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODMotif2/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODSST2/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODSST2.yaml 4 | -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODSST2/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODSST2/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODTwitter/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../GOODTwitter.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODTwitter/length/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/GOODTwitter/length/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/LBAPcore/assay/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/LBAPcore/assay/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/LBAPcore/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | - ../../LBAPcore.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/LBAPcore/scaffold/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/LBAPcore/scaffold/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/LBAPcore/size/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/LBAPcore/size/covariate/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | #lr: 4 | # - 1e-3 -------------------------------------------------------------------------------- /configs/sweep_configs/VREx/base.yaml: -------------------------------------------------------------------------------- 1 | includes: 2 | - ../base.yaml 3 | ood_param: 4 | - 1e3 5 | - 100 6 | - 10 7 | - 1 -------------------------------------------------------------------------------- /configs/sweep_configs/base.yaml: -------------------------------------------------------------------------------- 1 | exp_round: 2 | - 1 3 | - 2 4 | - 3 5 | log_file: 6 | - exp_0 7 | lr: 8 | - 1e-3 9 | - 1e-4 10 | weight_decay: 11 | - 0. 12 | - 1e-3 13 | - 1e-4 14 | # - 1e-5 15 | clean_save: # Comment it to make it false. Att: Setting it to false won't take effects. 16 | - true -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- 1 | name: pygood 2 | channels: 3 | - pytorch 4 | - conda-forge 5 | - defaults 6 | dependencies: 7 | - python=3.8 8 | - torchvision==0.11.2 9 | - cudatoolkit=11.3 10 | - torchaudio==0.10.1 11 | - pytorch==1.10.1 12 | - rdkit>=2020.09.5 13 | - pynvml=11.4 14 | - pip>=21.0 15 | - pip: 16 | - torch-scatter==2.0.9 17 | - torch-sparse==0.6.13 18 | - torch-geometric==2.0.4 19 | - cilog==1.2.3 20 | - gdown==4.4.0 21 | - matplotlib==3.5.2 22 | - munch==2.5.0 23 | - networkx==2.8 24 | - ogb>=1.3.3 25 | - pytest==7.1.2 26 | - ruamel.yaml==0.17.21 27 | - protobuf==3.20.1 28 | - sphinx>=4.5 29 | - sphinx-rtd-theme==1.0.0 30 | - tensorboard==2.8.0 31 | - tqdm==4.64.0 32 | - typed-argument-parser==1.7.2 33 | - dive-into-graphs 34 | --------------------------------------------------------------------------------