← all repositories

terrifyzhao/bert-utils

A Python utility library wrapping Google BERT for sentence embedding generation, text classification, and similarity computation.

1.7k stars Python Language ModelsML Frameworks
bert-utils
Not currently ranked — collecting fresh signals.
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.

Frequently asked

What is terrifyzhao/bert-utils?
A Python utility library wrapping Google BERT for sentence embedding generation, text classification, and similarity computation.
Is bert-utils open source?
Yes — terrifyzhao/bert-utils is open source, released under the Apache-2.0 license.
What language is bert-utils written in?
terrifyzhao/bert-utils is primarily written in Python.
How popular is bert-utils?
terrifyzhao/bert-utils has 1.7k stars on GitHub.
Where can I find bert-utils?
terrifyzhao/bert-utils is on GitHub at https://github.com/terrifyzhao/bert-utils.

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