Downloads · 30 days
5
1% of all-time downloads
distilbert-base-cross-encoder-first-p
brutusxu/distilbert-base-cross-encoder-first-p
distilbert-base-cross-encoder-first-p is a text classification model from brutusxu. Use it when you need a label for a piece of text. It is set up for transformers.
distilbert-base-uncased trained on MSMARCO Document Reranking task,
All-time downloads
589
Public
Repo size
536 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- distilbert
- Access
- Public
- Created
- Feb 5, 2023
- Updated
- Feb 5, 2023
- SHA
- a4adfa2f
Try a prompt
Model card
From the Hugging Face model README
distilbert-base-uncased trained on MSMARCO Document Reranking task,
usage
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained('brutusxu/distilbert-base-cross-encoder-first-p')
model = AutoModelForSequenceClassification.from_pretrained('brutusxu/distilbert-base-cross-encoder-first-p')
query = 'I love New York'
document = 'I like New York'
input = '<P>' + query + tokenizer.sep_token + '<Q>' + document
tokenized_input = tokenizer(input, return_tensors='pt')
ranking_score = model(**tokenized_input)
performance
on MSMARCO Document Reranking w. top-100 documents from BM25
MRR@10: 0.373
MRR@100: 0.381
nDCG@10: 0.442
nDCG@10: 0.475
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- Created
- Feb 5, 2023
- Updated
- Feb 5, 2023