NLP to classify news articles into catagories. Model inputs a news article as raw text and outputs one of the following categories: "business", "entertainment", "sport", "tech", or "politics".
BERT_Classifier_Demo.mov
git clone https://github.com/iansnyder333/BERT_News_Classifier.git
cd BERT_News_Classifier
python3.11 -m venv venv
source venv/bin/activate
pip3.11 install -r requirements.txt
python3.11 main.py
Python 3.11 recommended, the app was not tested on any other versions so performance is not known.
Model was trained with 5 EPOCHS using bbc-text csv file from kaggle. Model has 97% accuracy on testing data but is still in development for accuratley classifying new articles.