pyTorchTree

pyTorchTree copied to clipboard

PyTorch implementation of

1. RNN: Recursive Neural Network from https://nlp.stanford.edu/~socherr/EMNLP2013_RNTN.pdf
   python RecursiveNN.py
   
2. 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