google-research / tapas

End-to-end neural table-text understanding models.
Apache License 2.0
1.13k stars 216 forks source link

Finding Cell selection confidence #157

Closed Dhimant404 closed 1 year ago

Dhimant404 commented 2 years ago

Hello! Does the model output a confidence score for cell selection/output confidence? If so could you please point out the code that does that so it is possible to filter the output more precisely for custom cases. I have looked at the code and couldn't find the section that had cell selection/output confidence. I'm using tapas-base-finetuned-wtq right now and want to find the same, cell selection confidence/output confidence. Is there any part of the coder I could look at/ modify to get the confidence of the prediction? I'd appreciate any help, thanks in advance.