multiclass-classification

Repository for My HuggingFace Natural Language Processing Projects

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.

The model is based on a YOLOv8 (Deep learning Neural network architecture) and is trained on the publicly available dataset, which consists of lung CT scans of patients with and without lung cancer.