← all repositories

CyberZHG/keras-bert

A Keras implementation of BERT for loading pre-trained models, tokenization, feature extraction, and text classification.

2.4k stars Python Language ModelsML Frameworks
keras-bert
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

This repository provides a Keras-based implementation of BERT (Bidirectional Encoder Representations from Transformers). It enables loading official pre-trained BERT checkpoints for feature extraction and prediction tasks. The library includes a tokenizer for processing text into model-compatible indices, supports TPU execution, and offers utilities for fine-tuning BERT on classification and other NLP tasks.

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