Toolkit independent tools for visualizing NMT systems
Switch branches/tags
Nothing to show
Clone or download
Pull request Compare This branch is 28 commits ahead, 2 commits behind OpenNMT:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
attention_viz
beam_search_viz
.gitignore
LICENSE
README.md
attention_vis.py
generate_beam_viz.py
generate_embeddings_viz.py
requirements.txt
utils.py

README.md

VisTools

Toolkit independent tools for visualizing NMT systems

Installation

pip install -r requirements.txt

Beam Search Visualization

In sockeye use option --output-type beam_store to save the beam history in a json file.

python generate_beam_viz.py -d path/to/json/file
                            -o path/to/output/directory