/pytorch_text_classification_conv1d

This code is based on the paper by Yoon Kim, 2014 - 'Convolutional Neural Networks for Sentence Classification'.

Primary LanguagePython

Convolutional Neural Networks for Text Classification

This code is based on the paper by Yoon Kim, 2014 - 'Convolutional Networks for Sentence Classification'. The dependencies are pytorch, torchtext. This implementation uses FastText embeddings instead of Word2Vec ones, but obtains a comparable accuracy on the IMDB movie review dataset, which is similar to the one from that paper.

Please allow for 6.6GB download for the embeddings; it will occur automatically when running main.py.