https://github.com/harvardnlp/annotated-transformer

An annotated implementation of the Transformer paper.
https://github.com/harvardnlp/annotated-transformer

annotated notebook python

Last synced: 27 days ago
JSON representation

An annotated implementation of the Transformer paper.

• Host: GitHub
• URL: https://github.com/harvardnlp/annotated-transformer
• Owner: harvardnlp
• License: mit
• Created: 2018-03-21T16:23:15.000Z (over 6 years ago)
• Default Branch: master
• Last Pushed: 2024-04-07T09:58:46.000Z (7 months ago)
• Last Synced: 2024-10-01T13:24:13.229Z (about 1 month ago)
• Topics: annotated, notebook, python
• Language: Jupyter Notebook
• Homepage: http://nlp.seas.harvard.edu/annotated-transformer
• Size: 17.8 MB
• Stars: 5,613
• Watchers: 66
• Forks: 1,215
• Open Issues: 29
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
Code for The Annotated Transformer blog post:

http://nlp.seas.harvard.edu/annotated-transformer/

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/harvardnlp/annotated-transformer/blob/master/AnnotatedTransformer.ipynb)

![image](https://user-images.githubusercontent.com/35882/166251887-9da909a9-660b-45a9-ae72-0aae89fb38d4.png)

# Package Dependencies

Use `requirements.txt` to install library dependencies with pip:

```

pip install -r requirements.txt

```

# Notebook Setup

The Annotated Transformer is created using [jupytext](https://github.com/mwouts/jupytext).

Regular notebooks pose problems for source control - cell outputs end up in the repo history and diffs between commits are difficult to examine. Using jupytext, there is a python script (`.py` file) that is automatically kept in sync with the notebook file by the jupytext plugin.

The python script is committed contains all the cell content and can be used to generate the notebook file. The python script is a regular python source file, markdown sections are included using a standard comment convention, and outputs are not saved. The notebook itself is treated as a build artifact and is not commited to the git repository.

Prior to using this repo, make sure jupytext is installed by following the [installation instructions here](https://github.com/mwouts/jupytext/blob/main/docs/install.md).

To produce the `.ipynb` notebook file using the markdown source, run (under the hood, the `notebook` build target simply runs `jupytext --to ipynb the_annotated_transformer.py`):

```

make notebook

```

To produce the html version of the notebook, run:

```

make html

```

`make html` is just a shortcut for for generating the notebook with `jupytext --to ipynb the_annotated_transformer.py` followed by using the jupyter nbconvert command to produce html using `jupyter nbconvert --to html the_annotated_transformer.ipynb`                             

 

# Formatting and Linting

To keep the code formatting clean, the annotated transformer git repo has a git action to check that the code conforms to [PEP8 coding standards](https://www.python.org/dev/peps/pep-0008/).

To make this easier, there are two `Makefile` build targets to run automatic code formatting with black and flake8.

Be sure to [install black](https://github.com/psf/black#installation) and [flake8](https://flake8.pycqa.org/en/latest/).

You can then run:

```

make black

```

(or alternatively manually call black `black --line-length 79 the_annotated_transformer.py`) to format code automatically using black and:

```

make flake

```

(or manually call flake8 `flake8 --show-source the_annotated_transformer.py) to check for PEP8 violations.

It's recommended to run these two commands and fix any flake8 errors that arise, when submitting a PR, otherwise the github actions CI will report an error.