Skip to content

TanyaChutani/Text-Multilabel-Classification-BERT-Tf2.x

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Text-Multilabel-Classification-BERT-Tf2.0

This repository contains a Tensorflow2.0 implementation of the pretrained BERT model for multi-label text classification.

Steps

  1. Download data from kaggle
  2. Download pretrained model weights with the help of download_bert.sh
  3. Run train_bert.py