Text Classification with DistilBERT This repository contains a Python implementation for text classification using DistilBERT. The project demonstrates how to preprocess text data, and fine-tune a DistilBERT model for classification tasks. The dataset used includes various categories such as general,procedure,diagnosis,etc. Additionally, the project covers how to introduce perturbations to the input text using TextAttack library, performing advrsarial attack and training, and evaluating model performance with metrics such as accuracy and precision.
-
Notifications
You must be signed in to change notification settings - Fork 0
SameehanNS/Text_classifier_textattack
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Medical dialogue classifier using DistilBERT model, implementing TextAttack along with adversarial training.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published