← all repositories

xuyige/BERT4doc-Classification

A research implementation investigating optimal methods for fine-tuning BERT on text classification.

641 stars Python Language ModelsML Frameworks
BERT4doc-Classification
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository contains the code and experiments for a paper studying how to fine-tune BERT for text classification tasks. The work explores different fine-tuning strategies and hyperparameters for BERT, a pre-trained language model, when adapting it for downstream classification tasks. It supports multiple datasets including Sogou News and other text classification benchmarks, and provides checkpoints compatible with the fastNLP library’s BertEmbedding module.

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