PyTorch implementation of
- RNN: Recursive Neural Network from https://nlp.stanford.edu/~socherr/EMNLP2013_RNTN.pdf
python RecursiveNN.py - TreeLSTM from https://arxiv.org/abs/1503.00075
python TreeLSTM.py
add cuda as an argument to run it using cuda
Requirements:
nltk
pytorch
progressbar
Mixed code from:
Socher's cs224d class (see for e.g. https://github.com/kingtaurus/cs224d/tree/master/assignment3)
and https://gist.github.com/wolet/1b49c03968b2c83897a4a15c78980b18