BertViz

Tool for visualizing BERT's attention layers. Extends Tensor2Tensor visualization tool.

Jupyter Notebooks

"Map" of BERT [new]:

Attention detail view:

Original summary view:

This project is licensed under the Apache 2.0 License - see the LICENSE file for details

This project incorporates code from the following repos:

Name		Name	Last commit message	Last commit date
Latest commit History 80 Commits
bertviz		bertviz
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
bertviz_detail.ipynb		bertviz_detail.ipynb
bertviz_map.ipynb		bertviz_map.ipynb
bertviz_summary.ipynb		bertviz_summary.ipynb
requirements.txt		requirements.txt