Skip to content

Using TF BERT Transformer model for multi-class text classification

Notifications You must be signed in to change notification settings

aditya10/TensorFlow-BERT-Text-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

TensorFlow-BERT-Text-Classification

Using TF BERT Transformer model for multi-class text classification

Usage

This notebook is intented to run on Google Colab. Setup for importing the dataset is documented in the first section of my blog post: Using FastAI’s ULMFiT to make a state-of-the-art multi-class text classifier

Resources

Dataset: https://github.com/jacoxu/StackOverflow

This code is partially taken from: https://colab.research.google.com/github/google-research/bert/blob/master/predicting_movie_reviews_with_bert_on_tf_hub.ipynb

About

Using TF BERT Transformer model for multi-class text classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published