/CNN-Text-Classification

reproduce CNN of kim, 2014 on pytorch

Primary LanguageShell

CNN-Text-Classification

Implementing CNN for Movie Review accoding to Kim (2014).

Prepare data

python process_data.py <path_to_word2vec>

Start Training

static

python main.py --embedding_freeze

Acc=0.8098

non-static

python main.py

Acc=0.8141

Dependencies

  • pytorch==0.3.0
  • python3
  • numpy