yezhejack / CNN-Text-Classification

reproduce CNN of kim, 2014 on pytorch

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