Simple Transformers — Multi-Class Text Classification with BERT, RoBERTa, XLNet, XLM, and DistilBERT

Simple Transformers is the “it just works” Transformer library. If you are looking to use Transformers for real applications (in 3 lines of code), without worrying about the technical details, this is for you.

Preface

The Simple Transformers library is built on top of the excellent Transformers library by Hugging Face. The Hugging Face Transformers library is the library for researchers and other people who need extensive control over how things are done. It is also the best choice when you need to stray off the beaten path, do things differently, or do new things altogether. Simple Transformers is, well, a lot simpler.

Introduction

Simple Transformers is designed for when you need to get something done and you want it done now. No mucking about with source code, no hours of hair-pulling while trying to figure out how to even set the damn thing up. Text classification is so common that it should be easy, right? Simple Transformers thinks so and is here to do exactly that!