Downloads · 30 days
14
17% of all-time downloads
LilaBoualili/electra-sim-pair
electra-sim-pair 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 using the Sim-Pair marking strategy that highlights exact term matches between the q…
Downloads · 30 days
14
17% of all-time downloads
All-time downloads
81
Public
Repo size
1.3 GB
Likes
0
Public
Click a slice to open those files.
.h5438 MB · 50%
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 using the Sim-Pair marking strategy that highlights exact term matches between the query and the passage via marker tokens (#). 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.