Skip to content

Tensorflow implementation of some text classification models

License

Notifications You must be signed in to change notification settings

bhsimon0810/text-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text Classification

Tensorflow implementation of some text classification models.

Dependencies

  • Python 3.6
  • Tensorflow == 1.14.0

Usage

Datasets

  • The datasets (imdb, yelp-13, yelp-14, yelp15) are collected and processed by Duyu Tang, and the resources can be downloaded here.
  • The glove word embedding.

Training & Evaluating

To preprocess the dataset (make sure all the data are in correct directories)

python data_helpers.py

To train the model (please check all the hyper-parameters)

python train.py

To evaluate the trained model

python eval.py

Models

fastText, CNN, RNN w/o attention, RCNN, GatedNN, HAN, DMN

About

Tensorflow implementation of some text classification models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages