tdhook#

tdhook

Interpreting torch Deep Neural Networks

tdhook is a package for explaining torch deep neural networks based on tensordict and torch hooks.

Key Features

_images/one.png

Efficiency

Memory and time efficient, no large bundle, only what you need.

_images/two.png

Interpretability

Wide range of interpretability methods ready to be plugged in, with a general API to have full control over the interpretability process.