← all repositories

brightmart/text_classification

A repository of deep learning text classification models including TextCNN, TextRNN, and support for BERT/ALBERT fine-tuning.

7.9k stars Python Language ModelsML Frameworks
text_classification
Velocity · 7d
+2.4
★ / day
Trend
steady
star history

This repository provides baseline models for text classification tasks in NLP using deep learning. It includes implementations of classic architectures like TextCNN, TextRNN, and integrates with pre-trained models such as BERT and ALBERT. The project supports both multi-class and multi-label classification scenarios, with models trained using TensorFlow.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.