BertViz

Définition

BertViz (abréviation de BERT Visualization) est un nom propre désignant un logiciel libre en Python pour visualiser l'attention dans un modèle de réseau neuronal autoattentif (Transformer).

Compléments

Il supporte la plupart des modèles de la bibliothèque Transformers (BERT, GPT-2, XLNet, RoBERTa, XLM, CTRL, MarianMT, etc.). Il étend l'outil de visualisation Tensor2Tensor de Llion Jones et la bibliothèque de transformateurs de HuggingFace.

Français

BertViz

Anglais

BertViz

Source : github