Skip to content

DucLeTrong/vietnamese-text-classify

Repository files navigation

Vietnamese text classifier using LSTM model

Accuracy

About data set

The Vietnamese dataset consists of 10 different topics

Training

  • Install requirement
>> pip install -r requirements.txt
  • Prepare data:
>> cd data/ && unrar x -r Test_Full.rar && unrar x -r Train_Full.rar

>> python load_data.py --data_path='data/'
  • Train model
>> python train.py --data_path="data/" --model_path="model/"

Results

png

  • Classify text
>> python infer.py --prime "Đêm hôm qua, đội tuyển Việt Nam đã bay đến Trung Quốc chuẩn bị cho giải vô địch Châu Á."

About

A deep learning model for classifying Vietnamese document according to topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published