tensorspace/Information-Bottleneck-for-Deep-Learning

Implementation of MLP (python) and CNN (PyTorch) with Information Plane visualization.

Jupyter Notebook