Pretrain model to classify text as yes, no, not sure

Models
1

hi,

do we have any pretrained model to classify text as yes, no, not sure?

any help is appreciated

thanks

2

Hi, I think any binary pretrained classifier can do it. We just need probability output (already have from dense with sigmoid activation)
Then, we can set our “unsure” threshold based on data (e.g. if prob is between 0.4-0.6 assign as unsure).

3

You could also try using a zero shot classifier here.

4

yes i already tried zero-shot and its working great :slightly_smiling_face:

thanks

1 Like