Fairseq Integration

Fairseq is a popular sequence modeling toolkit developed by Facebook AI Research. VizSeq can directly import and analyze model predictions generated by fairseq-generate or fairseq-interactive in Jupyter Notebook. The APIs are almost the same as the normal Jupyter Notebook APIs:

from vizseq.ipynb import fairseq_viz as vizseq_fs
log_path = 'examples/data/wmt14_fr_en_test.fairseq_generate.log'
vizseq_fs.view_stats(log_path)
vizseq_fs.view_examples(log_path, ['bleu', 'meteor'], need_g_translate=True)
vizseq_fs.view_scores(log_path, ['bleu', 'meteor'])
vizseq_fs.view_n_grams(log_path)
Copy