ThilinaRajapakse/pytorch-transformers-classification
Based on the Pytorch-Transformers library by HuggingFace. To be used as a starting point for employing Transformer models in text classification tasks. Contains code to easily train BERT, XLNet, RoBERTa, and XLM models for text classification.
This tool helps machine learning engineers and researchers classify text documents by fine-tuning powerful Transformer models like BERT, XLNet, and XLM. You provide your text data, and it outputs a trained model capable of categorizing new text. It serves as a foundational example for building advanced text classification systems.
312 stars. No commits in the last 6 months.
Use this if you are a machine learning engineer or researcher looking for a starting point to understand and implement Transformer models for text classification tasks with more control over the underlying process.
Not ideal if you need an easy-to-use, regularly maintained library with fewer low-level details for text classification; in that case, 'Simple Transformers' is recommended.
Stars
312
Forks
95
Language
Jupyter Notebook
License
Apache-2.0
Category
Last pushed
May 09, 2020
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/nlp/ThilinaRajapakse/pytorch-transformers-classification"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related tools
fidelity/textwiser
[AAAI 2021] TextWiser: Text Featurization Library
RandolphVI/Multi-Label-Text-Classification
About Muti-Label Text Classification Based on Neural Network.
ntumlgroup/LibMultiLabel
A library for multi-class and multi-label classification
xuyige/BERT4doc-Classification
Code and source for paper ``How to Fine-Tune BERT for Text Classification?``
allenai/scibert
A BERT model for scientific text.