Multi Class Text Classification (emotion detection) using 🤗transformers.
This repository demonstrates how to train an NLP model using transformers and datasets library together with a custom Amazon sagemaker-sdk extension to fine-tune a pre-trained transformer model (Roberta) on Multi Class Text Classification.
You can access the fine tuned model at the [hub]: https://huggingface.co/Jorgeutd/sagemaker-roberta-base-emotion and test it yourself.
Framework versions
- Transformers 4.12.3
- Pytorch 1.9.1
- Datasets 1.15.1
- Tokenizers 0.10.3
- AWS Sagemaker