Is it possible to get the label probablities before final prediction for binary classification?

utterworks / fast-bert

Super easy library for BERT based NLP models

Apache License 2.0

1.85k stars 342 forks source link

Is it possible to get the label probablities before final prediction for binary classification? #248

Open NawelAr opened 4 years ago

NawelAr commented 4 years ago

Hello,

Is it possible to get the label probabilities before final prediction for binary classification? I need the probabilities in order to draw a roc curve.

thanks