Machine Learning based framework for classification of news data.
-
Feature Extraction - Gensim Doc2Vec
-
Model Training - sklearn Logistic Regression, Keras Neural Model
-
UI Building - DJango Framework.
-
Prediction accuracy on Test dataset: 97% (approx.)
For the ML model code, check news_classify.ipynb file