Using Machine Learning to Classify News Articles

This research paper delves into various approaches and techniques for classifying news articles into different categories using machine learning and which ones resulted in the best outcome. We tested various models for this classification task. We used 4 models in total, including two that we created ourselves with the library SciKit Learn, and the other two, we found on HuggingFace, which we fine-tuned for our specific task. The best-performing model for this task was one of the deep learning models (Mini LM) from HuggingFace, which had an accuracy of 75.42% out of the 10,000 articles it was tested on. This model was also one of the most diminutive models that we used for classifying news articles.