├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── examples ├── train_embedding.py ├── train_iris.py ├── train_mnist.py └── train_stacked_mnist.py ├── images └── TabNet.png ├── requirements.txt ├── setup.cfg ├── setup.py ├── tabnet ├── __init__.py ├── custom_objects.py ├── stacked_tabnet.py └── tabnet.py └── tests └── test_custom_objects.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include README.md 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/README.md -------------------------------------------------------------------------------- /examples/train_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/examples/train_embedding.py -------------------------------------------------------------------------------- /examples/train_iris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/examples/train_iris.py -------------------------------------------------------------------------------- /examples/train_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/examples/train_mnist.py -------------------------------------------------------------------------------- /examples/train_stacked_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/examples/train_stacked_mnist.py -------------------------------------------------------------------------------- /images/TabNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/images/TabNet.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [bdist_wheel] 2 | universal = 1 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/setup.py -------------------------------------------------------------------------------- /tabnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/tabnet/__init__.py -------------------------------------------------------------------------------- /tabnet/custom_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/tabnet/custom_objects.py -------------------------------------------------------------------------------- /tabnet/stacked_tabnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/tabnet/stacked_tabnet.py -------------------------------------------------------------------------------- /tabnet/tabnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/tabnet/tabnet.py -------------------------------------------------------------------------------- /tests/test_custom_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/titu1994/tf-TabNet/HEAD/tests/test_custom_objects.py --------------------------------------------------------------------------------