Skip to content

Medical dialogue classifier using DistilBERT model, implementing TextAttack along with adversarial training.

Notifications You must be signed in to change notification settings

SameehanNS/Text_classifier_textattack

Repository files navigation

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.

About

Medical dialogue classifier using DistilBERT model, implementing TextAttack along with adversarial training.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published