Visualizations of convolutional layers in NLP tasks
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore
Makefile
README.md
productionize.py
train.py
utils.py
visualize.py

README.md

These are the training scripts for a blog post that I wrote about using convolutional neural networks for NLP. These scripts can be run from the command line, or using the recipes in the Makefile. The latter will handle installing the correct dependencies; it is assumed that your system has a GPU for training the models, so tensorflow-gpu is installed. You can change the TENSORFLOW_PIP_VERSION in the constants to tensorflow if you want to train without a GPU.

make binary   ; make productionize  # Creates the binary classification examples
make language ; make productionize  # Creates the language model examples
make clean                          # Removes any created directories