Skip to content

Apply CNN-LSTM model on multi-class text classification task.

Notifications You must be signed in to change notification settings

CH-YYK/cnn-lstm-text-classification

Repository files navigation

CNN-LSTM-Text-Classification

Apply CNN-LSTM model on multi-class text classification task.

GRAPH

Instructions

After setting the model configerations in train.py run the following commands in terminal.

$ python train.py

and view the result in tensorboard (you could replace the path after --logdir by yours).

$ tensorboard --logdir='1536427044/runs/summaries'

Performance

accuracy loss

Discussion

still faced with the problem of overfitting. I have tried l2 regularization and dropout, cannot see a good improvement. If you guys who view this repo and has alternatives, leave comments here.

About

Apply CNN-LSTM model on multi-class text classification task.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages