Classify Text/Document in different classes using pre-trained BERT encoder.
Classify the given text files into the following classes.
- Anaconda or Miniconda installed
- Git installed
Follow these steps to set up and run your project.
-
Create a new virtual environment using conda:
conda create -n classification python
-
Activate the virtual environment:
conda activate classifcation
-
Clone the repository
git clone https://github.com/shulavkarki/Text-Classifcation-using-BERT
-
Navigate to the project repository:
cd Text-Classifcation-using-BERT
-
Install the required packages, libraries and frameworks:
pip install -r requirements.txt
-
Training:
python train.py
-
Evaluation:
python evaluate.py
Change the config for different training configuration according to your device.