├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── assets ├── distilled │ ├── checkpoint │ ├── tree-model.data-00000-of-00001 │ ├── tree-model.index │ └── tree-model.meta ├── img │ ├── branching.png │ ├── deeper.jpg │ ├── epoch.gif │ ├── infer.gif │ └── sample.gif ├── nn-model.hdf5 └── non-distilled │ ├── checkpoint │ ├── tree-model.data-00000-of-00001 │ ├── tree-model.index │ └── tree-model.meta ├── makegif.sh ├── mnist.ipynb ├── models ├── __init__.py ├── convnet.py ├── tree.py └── utils.py ├── requirements.txt └── requirements_gpu.txt /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/README.md -------------------------------------------------------------------------------- /assets/distilled/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/distilled/checkpoint -------------------------------------------------------------------------------- /assets/distilled/tree-model.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/distilled/tree-model.data-00000-of-00001 -------------------------------------------------------------------------------- /assets/distilled/tree-model.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/distilled/tree-model.index -------------------------------------------------------------------------------- /assets/distilled/tree-model.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/distilled/tree-model.meta -------------------------------------------------------------------------------- /assets/img/branching.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/img/branching.png -------------------------------------------------------------------------------- /assets/img/deeper.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/img/deeper.jpg -------------------------------------------------------------------------------- /assets/img/epoch.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/img/epoch.gif -------------------------------------------------------------------------------- /assets/img/infer.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/img/infer.gif -------------------------------------------------------------------------------- /assets/img/sample.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/img/sample.gif -------------------------------------------------------------------------------- /assets/nn-model.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/nn-model.hdf5 -------------------------------------------------------------------------------- /assets/non-distilled/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/non-distilled/checkpoint -------------------------------------------------------------------------------- /assets/non-distilled/tree-model.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/non-distilled/tree-model.data-00000-of-00001 -------------------------------------------------------------------------------- /assets/non-distilled/tree-model.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/non-distilled/tree-model.index -------------------------------------------------------------------------------- /assets/non-distilled/tree-model.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/assets/non-distilled/tree-model.meta -------------------------------------------------------------------------------- /makegif.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/makegif.sh -------------------------------------------------------------------------------- /mnist.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/mnist.ipynb -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/convnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/models/convnet.py -------------------------------------------------------------------------------- /models/tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/models/tree.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmartak/distill-nn-tree/HEAD/models/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tensorflow>=1.10.0 2 | matplotlib>=2.1.0 3 | -------------------------------------------------------------------------------- /requirements_gpu.txt: -------------------------------------------------------------------------------- 1 | tensorflow-gpu>=1.10.0 2 | matplotlib>=2.1.0 3 | --------------------------------------------------------------------------------