/adverse-drug-effect-detection-transformers-sagemaker

Train an NLP model using transformers and datasets library together with a custom Amazon sagemaker-sdk to detect adverse drugs effects

Primary LanguageJupyter Notebook

adverse-drug-effect-detection using using 🤗transformers and sagemaker

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 to detect adverse drug effects.

Framework versions

  • Transformers 4.12.3
  • Pytorch 1.9.1
  • Datasets 1.15.1
  • Tokenizers 0.10.3
  • AWS Sagemaker

Credits

Credits to https://github.com/marshmellow77 on the sagamaker DLC guidance.