• Stars
    star
    105
  • Rank 328,196 (Top 7 %)
  • Language
    Python
  • Created almost 7 years ago
  • Updated over 6 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Convolutional Neural Networks for Sentence Classification in PyTorch

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