← all repositories

hellonlp/classifier-multi-label

Multi-label text classification toolkit implementing BERT, TextCNN, dense, and seq2seq approaches in TensorFlow.

805 stars Python Language ModelsML Frameworks
classifier-multi-label
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

This repository provides implementations for multi-label text classification using multiple neural architectures. It supports BERT-based classification using the [CLS] token embedding, TextCNN on BERT outputs, dense layers, and seq2seq models. The library uses sigmoid cross-entropy loss for multi-label prediction and includes utilities for handling variable-length label assignments.

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