Skip to content

Classify Text/Document in different classes using pre-trained BERT encoder.

Notifications You must be signed in to change notification settings

shulavkarki/Text-Classifcation-using-BERT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text Classifcation using BERT

Classify Text/Document in different classes using pre-trained BERT encoder.

Classify the given text files into the following classes.

Prerequisites

  • Anaconda or Miniconda installed
  • Git installed

Getting Started

Follow these steps to set up and run your project.

Installation

  1. Create a new virtual environment using conda:

    conda create -n classification python
    
  2. Activate the virtual environment:

    conda activate classifcation
    
  3. Clone the repository

    git clone https://github.com/shulavkarki/Text-Classifcation-using-BERT
    
  4. Navigate to the project repository:

    cd Text-Classifcation-using-BERT
    
    
  5. Install the required packages, libraries and frameworks:

    pip install -r requirements.txt
    
  6. Training:

    python train.py
    
  7. Evaluation:

    python evaluate.py
    
    

Note

Change the config for different training configuration according to your device.

About

Classify Text/Document in different classes using pre-trained BERT encoder.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages