Downloads · 30 days
13
2% of all-time downloads
electra-vanilla
LilaBoualili/electra-vanilla
electra-vanilla is a text classification model from LilaBoualili. Use it when you need a label for a piece of text. It is set up for transformers.
At its core it uses an ELECTRA-Base model (google/electra-base-discriminator) fine-tuned on the MS MARCO passage classification task. It can be loaded using the TF/AutoModelForSequenceClassification classes but it fol…
All-time downloads
847
Public
Repo size
1.3 GB
Likes
0
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
- May 18, 2021
- SHA
- 92aa82dd
Try a prompt
Model card
From the Hugging Face model README
At its core it uses an ELECTRA-Base model (google/electra-base-discriminator) fine-tuned on the MS MARCO passage classification task. It can be loaded using the TF/AutoModelForSequenceClassification classes but it follows the same classification layer defined for BERT similarly to the TFElectraRelevanceHead in the Capreolus BERT-MaxP implementation.
Refer to our github repository for a usage example for ad hoc ranking.
- Task
- Text Classification
- Library
- transformers
- Type
- electra
- Created
- Mar 2, 2022
- Updated
- May 18, 2021