Views
No views yet
facebook/dpr-question_encoder-single-nq-base| K | Recall@K |
|---|---|
| 1 | 0.5720 |
| 5 | 0.7420 |
| 10 | 0.7840 |
| 20 | 0.8090 |
1from transformers import DPRQuestionEncoder, DPRQuestionEncoderTokenizer
2
3tokenizer = DPRQuestionEncoderTokenizer.from_pretrained("Anshul017/factuality-dpr-question")
4model = DPRQuestionEncoder.from_pretrained("Anshul017/factuality-dpr-question")
5
6query = "Where was Marie Curie born?"
7enc = tokenizer(query, return_tensors="pt")
8vec = model(**enc).pooler_output # [1, 768]
9