Skip to content

Convolutional Neural Networks for Sentence Classification in PyTorch

Notifications You must be signed in to change notification settings

Cheneng/TextCNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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			

About

Convolutional Neural Networks for Sentence Classification in PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages