capreolus.reranker.TFVanillaBert

Module Contents

Classes

TFVanillaBert_Class(extractor, config, *args, **kwargs)
TFVanillaBERT(config=None, provide=None, share_dependency_objects=False, build=True)	TensorFlow implementation of Vanilla BERT.

capreolus.reranker.TFVanillaBert.logger

class capreolus.reranker.TFVanillaBert.TFVanillaBert_Class(extractor, config, *args, **kwargs)

Bases: tensorflow.keras.Model

call(self, x, **kwargs)

During training, both posdoc and negdoc are passed During eval, both posdoc and negdoc are passed but negdoc would be a zero tensor Whether negdoc is a legit doc tensor or a dummy zero tensor is determined by which sampler is used (eg: sampler.TrainDataset) as well as the extractor (eg: EmbedText)

class capreolus.reranker.TFVanillaBert.TFVanillaBERT(config=None, provide=None, share_dependency_objects=False, build=True)

Bases: capreolus.reranker.Reranker

TensorFlow implementation of Vanilla BERT.

module_name = TFVanillaBERT

dependencies

config_spec

build_model(self)