GitHub - spartan-minhbui/sentiment-analysis: Pipeline for sentiment classification, use trainer of HuggingFace and use ONNX to export model

Finetune pretrained model for Sentiment analysis

Try some pretrained model:
- PhoBERT: vinai/phobert-base-v2
- Bloom: bigscience/bloom-560m
Architecture: XXXXForSequenceClassification
Use Trainer of Huggingface to training model

Export environment variables: while read LINE; do export "$LINE"; done < .env
Run training: PRETRAINED_PATH=bigscience/bloom-560m CUDA_VISIBLE_DEVICES=1,2 python -m torch.distributed.launch --nproc_per_node 2 --master-port=30000 pipeline/trainer.py

Name		Name	Last commit message	Last commit date
Latest commit History 25 Commits
common		common
pipeline		pipeline
.gitignore		.gitignore
README.md		README.md
requirements.txt		requirements.txt
sentiment-analysis-using-roberta-lstm.ipynb		sentiment-analysis-using-roberta-lstm.ipynb
setup_source_dir.py		setup_source_dir.py