/TextCNN

Convolutional Neural Networks for Sentence Classification in PyTorch

Primary LanguagePython

Convolutional Neural Networks for Sentence Classification

This repo implements the Convolutional Neural Networks for Sentence Classification (Yoon Kim) using PyTorch

model_archi

You should rewrite the Dataset class in the data/dataset.py
and put your data in '/data/train' or any other directory.

run by

python3 main.py --lr=0.01 --epoch=20 --batch_size=16 --gpu=0 --seed=0 --label_num=2