terrifyzhao/bert-utils
A Python utility library wrapping Google BERT for sentence embedding generation, text classification, and similarity computation.

Velocity · 7d
+0.6
★ / day
Trend
→steady
star history
This repository simplifies Google BERT for common NLP tasks. It provides a BertVector class that generates sentence embeddings from pretrained Chinese BERT models without fine-tuning, and a BertSim class for fine-tuning-based text classification and similarity tasks. Users load a pretrained Chinese BERT checkpoint and call encode with text lists to obtain embeddings, or train classifiers on their own labeled datasets.