The Vietnamese dataset consists of 10 different topics
- 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/"
- 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 Á."
