https://github.com/tanyachutani/text-multilabel-classification-bert-tf2.x
Text-Multilabel-Classification-BERT-Tf2.0
https://github.com/tanyachutani/text-multilabel-classification-bert-tf2.x
bert-model multi-label-classification tensorflow text-classification tf2
Last synced: 7 months ago
JSON representation
Text-Multilabel-Classification-BERT-Tf2.0
- Host: GitHub
- URL: https://github.com/tanyachutani/text-multilabel-classification-bert-tf2.x
- Owner: TanyaChutani
- Created: 2021-03-18T18:33:37.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2021-04-08T18:36:10.000Z (over 4 years ago)
- Last Synced: 2025-01-10T11:59:31.650Z (9 months ago)
- Topics: bert-model, multi-label-classification, tensorflow, text-classification, tf2
- Language: Python
- Homepage:
- Size: 16.6 KB
- Stars: 2
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 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](https://www.kaggle.com/c/jigsaw-toxic-comment-classification-challenge/data)
2. Download pretrained model weights with the help of `download_bert.sh`
3. Run `train_bert.py`