cooelf / SemBERT

Semantics-aware BERT for Language Understanding (AAAI 2020)
https://arxiv.org/abs/1909.02209
MIT License
285 stars 55 forks source link

Use your code for textual entailment #7

Closed mmmjung closed 4 years ago

mmmjung commented 4 years ago

Hi, thanks for the code!

Can I somehow use your code to get textual entailment probabilities for pairs of sentences from my own dataset? If yes, please help give me some hints.

cooelf commented 4 years ago

You can use the logits here: https://github.com/cooelf/SemBERT/blob/master/run_classifier.py#L1223