/Multiclass-Text-Classification-with-DistilBERT-on-COVID-19-Tweets

I implement a deep learning network to classify COVID-19 Tweets into 5 categories and 3 categories using DistilBERT (a lighter version of BERT) as an embedding layer along with an LSTM and Dense Layer. I Achieve 65% accuracy with 5 categories and 80% accuracy on 3 categories.

Primary LanguageJupyter Notebook

Stargazers