Skip to content

Latest commit

 

History

History
29 lines (16 loc) · 646 Bytes

README.md

File metadata and controls

29 lines (16 loc) · 646 Bytes

Text Classification

Multi-model implementation for text classification.

Commands

Install required fundamental packages:

pip3 install pandas
pip3 install torch

If you wish to go with the same dataset as here, follow the guidelines here.

Prepare your dataset with the following command:

python3 dataset/dataset.py

Train your model with:

python3 train.py

Configuration

You can always manage your project configuration using the file config.py.

Remember: This should be done before training your model or it may not work as expected.