search

yya518 / FinBERT

A Pretrained BERT Model for Financial Communications. https://arxiv.org/abs/2006.08097
Apache License 2.0
560 stars 128 forks source link

Is there a prediction script available? #6

Closed man0007 closed 4 years ago

man0007 commented 4 years ago

HI,

Congrats for the wonderful work.

I would much appreciate if you would construct a prediction script which could work like this:

python bert_prediction.py --input "Sentence for classification" -- weights (path to weights)

yya518 commented 4 years ago

We provide a Jupyter notebook script in the repo. It loads the pre-trained model and makes predictions on sentences.