├── LICENSE ├── README.md ├── bin ├── __init__.py ├── autoint.py ├── dcn2.py ├── mlp.py ├── saint.py └── tpberta_modeling.py ├── configs └── default │ ├── autoint │ └── cfg.json │ ├── catboost │ └── cfg.json │ ├── dcnv2 │ └── cfg.json │ ├── tabnet │ └── cfg.json │ └── xgboost │ └── cfg.json ├── lib ├── __init__.py ├── aux.py ├── data.py ├── data_utils.py ├── deep.py ├── env.py ├── feature_encoder.py ├── metrics.py ├── optim.py ├── saint_lib │ ├── __init__.py │ ├── augmentations.py │ ├── data_openml.py │ └── models │ │ ├── __init__.py │ │ ├── model.py │ │ ├── pretrainmodel.py │ │ └── pretrainmodel_vision.py └── util.py ├── requirements.txt └── scripts ├── clean_feat_names.py ├── examples ├── 0-pretrain.sh ├── 1-finetune_default.sh ├── 2-tune.sh └── 3-finetune_tuned.sh ├── finetune ├── default │ ├── run_default_config_tabnet.py │ ├── run_default_config_tpberta.py │ ├── run_default_config_transtab.py │ ├── run_default_config_tree.py │ └── run_default_config_xtab.py ├── tune │ ├── tune_dnns.py │ ├── tune_ftt.py │ └── tune_trees.py └── tuned │ ├── run_tuned_config_dnns.py │ ├── run_tuned_config_ftt.py │ └── run_tuned_config_tree.py └── pretrain └── pretrain_tpberta.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/README.md -------------------------------------------------------------------------------- /bin/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/bin/__init__.py -------------------------------------------------------------------------------- /bin/autoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/bin/autoint.py -------------------------------------------------------------------------------- /bin/dcn2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/bin/dcn2.py -------------------------------------------------------------------------------- /bin/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/bin/mlp.py -------------------------------------------------------------------------------- /bin/saint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/bin/saint.py -------------------------------------------------------------------------------- /bin/tpberta_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/bin/tpberta_modeling.py -------------------------------------------------------------------------------- /configs/default/autoint/cfg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/configs/default/autoint/cfg.json -------------------------------------------------------------------------------- /configs/default/catboost/cfg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/configs/default/catboost/cfg.json -------------------------------------------------------------------------------- /configs/default/dcnv2/cfg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/configs/default/dcnv2/cfg.json -------------------------------------------------------------------------------- /configs/default/tabnet/cfg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/configs/default/tabnet/cfg.json -------------------------------------------------------------------------------- /configs/default/xgboost/cfg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/configs/default/xgboost/cfg.json -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/__init__.py -------------------------------------------------------------------------------- /lib/aux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/aux.py -------------------------------------------------------------------------------- /lib/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/data.py -------------------------------------------------------------------------------- /lib/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/data_utils.py -------------------------------------------------------------------------------- /lib/deep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/deep.py -------------------------------------------------------------------------------- /lib/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/env.py -------------------------------------------------------------------------------- /lib/feature_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/feature_encoder.py -------------------------------------------------------------------------------- /lib/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/metrics.py -------------------------------------------------------------------------------- /lib/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/optim.py -------------------------------------------------------------------------------- /lib/saint_lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/saint_lib/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/saint_lib/augmentations.py -------------------------------------------------------------------------------- /lib/saint_lib/data_openml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/saint_lib/data_openml.py -------------------------------------------------------------------------------- /lib/saint_lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/saint_lib/models/__init__.py -------------------------------------------------------------------------------- /lib/saint_lib/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/saint_lib/models/model.py -------------------------------------------------------------------------------- /lib/saint_lib/models/pretrainmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/saint_lib/models/pretrainmodel.py -------------------------------------------------------------------------------- /lib/saint_lib/models/pretrainmodel_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/saint_lib/models/pretrainmodel_vision.py -------------------------------------------------------------------------------- /lib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/lib/util.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/clean_feat_names.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/clean_feat_names.py -------------------------------------------------------------------------------- /scripts/examples/0-pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/examples/0-pretrain.sh -------------------------------------------------------------------------------- /scripts/examples/1-finetune_default.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/examples/1-finetune_default.sh -------------------------------------------------------------------------------- /scripts/examples/2-tune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/examples/2-tune.sh -------------------------------------------------------------------------------- /scripts/examples/3-finetune_tuned.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/examples/3-finetune_tuned.sh -------------------------------------------------------------------------------- /scripts/finetune/default/run_default_config_tabnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/default/run_default_config_tabnet.py -------------------------------------------------------------------------------- /scripts/finetune/default/run_default_config_tpberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/default/run_default_config_tpberta.py -------------------------------------------------------------------------------- /scripts/finetune/default/run_default_config_transtab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/default/run_default_config_transtab.py -------------------------------------------------------------------------------- /scripts/finetune/default/run_default_config_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/default/run_default_config_tree.py -------------------------------------------------------------------------------- /scripts/finetune/default/run_default_config_xtab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/default/run_default_config_xtab.py -------------------------------------------------------------------------------- /scripts/finetune/tune/tune_dnns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/tune/tune_dnns.py -------------------------------------------------------------------------------- /scripts/finetune/tune/tune_ftt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/tune/tune_ftt.py -------------------------------------------------------------------------------- /scripts/finetune/tune/tune_trees.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/tune/tune_trees.py -------------------------------------------------------------------------------- /scripts/finetune/tuned/run_tuned_config_dnns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/tuned/run_tuned_config_dnns.py -------------------------------------------------------------------------------- /scripts/finetune/tuned/run_tuned_config_ftt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/tuned/run_tuned_config_ftt.py -------------------------------------------------------------------------------- /scripts/finetune/tuned/run_tuned_config_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/finetune/tuned/run_tuned_config_tree.py -------------------------------------------------------------------------------- /scripts/pretrain/pretrain_tpberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jyansir/tp-berta/HEAD/scripts/pretrain/pretrain_tpberta.py --------------------------------------------------------------------------------