← all repositories

lonePatient/Bert-Multi-Label-Text-Classification

A PyTorch implementation for fine-tuning BERT and XLNET models on multi-label text classification tasks.

923 stars Python Language Models
Bert-Multi-Label-Text-Classification
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides code for fine-tuning pretrained BERT and XLNET models on multi-label text classification problems. It uses the Hugging Face transformers library and implements standard deep learning training components including data preprocessing, model fine-tuning, callbacks, and evaluation. Users need to download pretrained weights and organize them in a specific directory structure before running the training scripts.

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