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

hi,

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

any help is appreciated

thanks

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).

You could also try using a zero shot classifier here.

yes i already tried zero-shot and its working great

thanks

1 Like