capreolus.extractor.birch_bertpassage

Module Contents

Classes

BirchBertPassage

Prepare and parse TF training feature that contain multiple passage per query.

Attributes

logger

capreolus.extractor.birch_bertpassage.logger[source]
class capreolus.extractor.birch_bertpassage.BirchBertPassage(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.extractor.common.MultipleTrainingPassagesMixin, capreolus.extractor.bertpassage.BertPassage

Prepare and parse TF training feature that contain multiple passage per query. That is, the “pos_bert_input” features prepared by extractor’s id2vec() function should have 3 dimension

module_name = birchbertpassage[source]
config_spec[source]
id2vec(qid, posid, negid=None, label=None, **kwargs)[source]

See parent class for docstring