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.