BERTFinanceNeg
BERTFinanceNeg copied to clipboard
BERTFinanceNeg
BERT-based Finance Negation Detection
Requirements
- Python 3.6
- PyTorch 1.0.0
- pytorch-pretrained-bert
Usage
- Train with command, optional arguments could be found in train.py
python train.py --model_name bert --batch_size 16 --save True
- Infer with infer.py
Model
An overview of the BERT-based baseline is given below

Credits
- For any issues or suggestions about this work, don't hesitate to create an issue or directly contact me via [email protected] !