xuyige/BERT4doc-Classification
A research implementation investigating optimal methods for fine-tuning BERT on text 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.