BERT_binary_text_classification
BERT_binary_text_classification copied to clipboard
How to predict a sentence if it is 0 or 1?
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.