Downloads · 30 days
17
0% of all-time downloads
electra-base-msmarco
Capreolus/electra-base-msmarco
electra-base-msmarco is a text classification model from Capreolus. Use it when you need a label for a piece of text. It is set up for transformers.
ELECTRA-Base model (google/electra-base-discriminator) fine-tuned on the MS MARCO passage classification task. It is intended to be used as a ForSequenceClassification model, but requires some modification since it co…
All-time downloads
5.7K
Public
Repo size
1.3 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.h5438 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- electra
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Sep 8, 2020
- SHA
- a973d297
Try a prompt
Model card
From the Hugging Face model README
capreolus/electra-base-msmarco
Model description
ELECTRA-Base model (google/electra-base-discriminator) fine-tuned on the MS MARCO passage classification task. It is intended to be used as a ForSequenceClassification model, but requires some modification since it contains a BERT classification head rather than the standard ELECTRA classification head. See the TFElectraRelevanceHead in the Capreolus BERT-MaxP implementation for a usage example.
This corresponds to the ELECTRA-Base model used to initialize PARADE (ELECTRA) in PARADE: Passage Representation Aggregation for Document Reranking by Li et al. It was converted from the released TFv1 checkpoint. Please cite the PARADE paper if you use these weights.
- Task
- Text Classification
- Library
- transformers
- Type
- electra
- Created
- Mar 2, 2022
- Updated
- Sep 8, 2020