narratives-of-war / topic-rnn

Implementation (in progress) of Dieng et al.'s TopicRNN intended to be used as a baseline and starting point.

MIT License

10 stars 4 forks source link

Closed dangitstam closed 6 years ago

dangitstam commented 6 years ago

Experiments + Debugging from Movie Review Data

Migration to PyTorch 0.4.0
Batching now done within documents
Term frequencies now come from the previous backpropagation-through-time portion
Now supports usage of GloVe vectors with flagging for whether embeddings are trainable
Vocabulary is now normalized by case