yya518/FinBERT

Is there a prediction script available?

man0007 opened this issue · 1 comments

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)

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