Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

keishinkickback/Pytorch-RNN-text-classification

Open more actions menu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RNN-based short text classification

  • This is for multi-class short text classification.
  • Model is built with Word Embedding, LSTM ( or GRU), and Fully-connected layer by Pytorch.
  • A mini-batch is created by 0 padding and processed by using torch.nn.utils.rnn.PackedSequence.
  • Cross-entropy Loss + Adam optimizer.
  • Support pretrained word embedding (GloVe).

Model

  • Embedding --> Dropout --> LSTM(GRU) --> Dropout --> FC.

Preprocessing

python preprocess.py

Training

  • The following command starts training. Run it with -h for optional arguments.
python main.py

About

Word Embedding + LSTM + FC

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

Morty Proxy This is a proxified and sanitized view of the page, visit original site.