├── .github └── workflows │ └── python-app.yml ├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── benchmarks ├── README.md ├── covertype.py └── covertype_opt_inf.py ├── data └── test │ └── covtype_sample.csv ├── examples └── train_mnist.py ├── images ├── tabnet.png └── virtual_bs_vs_gbn.png ├── local └── tuner_orig_dataset.py ├── requirements.txt ├── scripts └── install.sh ├── setup.py ├── tabnet ├── __init__.py ├── callbacks │ ├── __init__.py │ ├── lrfinder.py │ └── tensorboard.py ├── datasets │ ├── __init__.py │ └── covertype.py ├── models │ ├── __init__.py │ ├── classify.py │ ├── gbn.py │ ├── model.py │ ├── transformers.py │ └── utils.py ├── schedules │ ├── __init__.py │ └── decay_with_warmup.py └── utils.py └── tests ├── test_classify.py ├── test_custom_bn.py ├── test_dataset.py ├── test_lr_finder.py └── test_tabnet.py /.github/workflows/python-app.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/.github/workflows/python-app.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/benchmarks/README.md -------------------------------------------------------------------------------- /benchmarks/covertype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/benchmarks/covertype.py -------------------------------------------------------------------------------- /benchmarks/covertype_opt_inf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/benchmarks/covertype_opt_inf.py -------------------------------------------------------------------------------- /data/test/covtype_sample.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/data/test/covtype_sample.csv -------------------------------------------------------------------------------- /examples/train_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/examples/train_mnist.py -------------------------------------------------------------------------------- /images/tabnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/images/tabnet.png -------------------------------------------------------------------------------- /images/virtual_bs_vs_gbn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/images/virtual_bs_vs_gbn.png -------------------------------------------------------------------------------- /local/tuner_orig_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/local/tuner_orig_dataset.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/scripts/install.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/setup.py -------------------------------------------------------------------------------- /tabnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/__init__.py -------------------------------------------------------------------------------- /tabnet/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/callbacks/__init__.py -------------------------------------------------------------------------------- /tabnet/callbacks/lrfinder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/callbacks/lrfinder.py -------------------------------------------------------------------------------- /tabnet/callbacks/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/callbacks/tensorboard.py -------------------------------------------------------------------------------- /tabnet/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tabnet/datasets/covertype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/datasets/covertype.py -------------------------------------------------------------------------------- /tabnet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/models/__init__.py -------------------------------------------------------------------------------- /tabnet/models/classify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/models/classify.py -------------------------------------------------------------------------------- /tabnet/models/gbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/models/gbn.py -------------------------------------------------------------------------------- /tabnet/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/models/model.py -------------------------------------------------------------------------------- /tabnet/models/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/models/transformers.py -------------------------------------------------------------------------------- /tabnet/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/models/utils.py -------------------------------------------------------------------------------- /tabnet/schedules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/schedules/__init__.py -------------------------------------------------------------------------------- /tabnet/schedules/decay_with_warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/schedules/decay_with_warmup.py -------------------------------------------------------------------------------- /tabnet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tabnet/utils.py -------------------------------------------------------------------------------- /tests/test_classify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tests/test_classify.py -------------------------------------------------------------------------------- /tests/test_custom_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tests/test_custom_bn.py -------------------------------------------------------------------------------- /tests/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tests/test_dataset.py -------------------------------------------------------------------------------- /tests/test_lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tests/test_lr_finder.py -------------------------------------------------------------------------------- /tests/test_tabnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ostamand/tensorflow-tabnet/HEAD/tests/test_tabnet.py --------------------------------------------------------------------------------