AIbase
Product LibraryTool Navigation

pytorch-transformers-classification

Public

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.

Creat2019-09-03T13:57:33
Update2025-02-26T20:01:53
307
Stars
0
Stars Increase

Related projects