/Roberta-Emotion-Classification-Transformers

train an NLP model using transformers and datasets library together with a custom Amazon sagemaker-sdk

Primary LanguageJupyter Notebook

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