capreolus.reranker.TFBERTMaxP
¶
Module Contents¶
Classes¶
BERT-style ClassificationHead (i.e., out_proj only – no dense). See transformers.TFElectraClassificationHead |
|
TensorFlow implementation of BERT-MaxP. |
-
class
capreolus.reranker.TFBERTMaxP.
TFElectraRelevanceHead
(dropout, out_proj, *args, **kwargs)[source]¶ Bases:
tensorflow.keras.layers.Layer
BERT-style ClassificationHead (i.e., out_proj only – no dense). See transformers.TFElectraClassificationHead
-
class
capreolus.reranker.TFBERTMaxP.
TFBERTMaxP_Class
(extractor, config, *args, **kwargs)[source]¶ Bases:
tensorflow.keras.layers.Layer
-
class
capreolus.reranker.TFBERTMaxP.
TFBERTMaxP
(config=None, provide=None, share_dependency_objects=False, build=True)[source]¶ Bases:
capreolus.reranker.Reranker
TensorFlow implementation of BERT-MaxP.
Deeper Text Understanding for IR with Contextual Neural Language Modeling. Zhuyun Dai and Jamie Callan. SIGIR 2019. https://arxiv.org/pdf/1905.09217.pdf