ThilinaRajapakse/BERT_binary_text_classification

How to predict a sentence if it is 0 or 1?

moh-yani opened this issue · 0 comments

Dear @ThilinaRajapakse ,

We have trained and evaluated our model using your code. We have resulted the metrics MCC etc on the evaluation. Now, we want to try for predicting a sentence if it is 0 or 1 of its label. Just for example the sentence is:
"The food is good. Unfortunately the service is..."

The output of the prediction we want is "1".

To do so, what code should we write?

Thank you for your attention and cooperation.