Visualize BERT's attention mechanism with a user-friendly script. Input text with a masked token, predict the masked word, and generate attention diagrams to understand BERT's focus. Ideal for AI enthusiasts and NLP researchers.
python
nlp
natural-language-processing
ai
data-visualization
transformer
cs50
attention-mechanism
bert
harvard-cs50
masked-language-model
-
Updated
Sep 19, 2024 - Jupyter Notebook