Skip to content

Classifier of news articles via Machine Learning (word embedding and LSTM)

Notifications You must be signed in to change notification settings

marcoramponi/news_classifier

Repository files navigation

News Classifier

Classifier of news articles via Machine Learning (word embedding and LSTM).

Natural Language Processing models for document classification tasks.

I used a baseline model via a pre-trained 50 dimensional word-embedding representation and a LSTM model

Data: trained the two models on a BBC archive of 2225 news articles (BBC dataset, available on Kaggle).

The LSTM model reached 89% accuracy on the test set.

About

Classifier of news articles via Machine Learning (word embedding and LSTM)

Resources

Stars

Watchers

Forks

Packages

No packages published