Skip to content

Multi-Class Text Classification with a simple multi-layer perceptron (MLP) model

Notifications You must be signed in to change notification settings

csyezheng/text-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text classification

This code repository covers multi-category classification and multi-label classification, and each classification problem contains a variety of different algorithms.

Multiclass classification

Gather Data

Use the THUCNews dataset that contains the text of 740,000 news from the thuctc.

Training

python -m text_classification.train_ngram_model

Hyperparameter search

python -m text_classification.tune_ngram_model

Predicting

python -m text_classification.predict_ngram_model

Multi-label classification

Gather Data

Use the THUCNews dataset that contains the text of 740,000 news from the thuctc.

Training

python -m text_classification.train_sequence_model

or train sequence model with fine-tuned pre-trained embeddings

python -m text_classification.train_fine_tuned_sequence_model

Predicting

python -m text_classification.predict_sequence_model

About

Multi-Class Text Classification with a simple multi-layer perceptron (MLP) model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages