dsiervo / topnews

Topic classification of news articles using machine learning and deep learning

topnews

Overview

Jupyter notebooks for news category classification using BERT and NBSVM in Kaggle News Category Dataset.

• Data exploration and BERT model: topNews.ipynb
• NBSVM model: Other_models.ipynb
• Style analysis: Style.ipynb
• Authors analysis: Author.ipynb
• Weekday heatmap: ML-NLP.ipynb

Dependencies

All the necessary dependencies are in the first cells of each notebook. You can also run it in google colab.