BERTFinanceNeg icon indicating copy to clipboard operation
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

model

Credits

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