dpr_scale/transforms/dpr_transform.py [14:25]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(
        self,
        text_transform,
        num_positive: int = 1,  # currently, like the original paper only 1 is supported
        num_negative: int = 7,
        neg_ctx_sample: bool = True, # sample from negative list
        pos_ctx_sample: bool = False, # sample from positives list
        num_val_negative: int = 7,  # num negatives to use in validation
        num_test_negative = None,  # defaults to num_val_negative
        use_title: bool = False,  # use the title for context passages
        sep_token: str = " ",  # sep token between title and passage
        text_column: str = "text",  # for onbox
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



dpr_scale/transforms/dpr_transform.py [153:164]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(
        self,
        text_transform,
        num_positive: int = 1,
        num_negative: int = 7,
        neg_ctx_sample: bool = True,
        pos_ctx_sample: bool = False,
        num_val_negative: int = 7,  # num negatives to use in validation
        num_test_negative = None,  # defaults to num_val_negative
        use_title: bool = False,  # Not supported for now
        sep_token: str = " ",  # sep token between question and passage
        text_column: str = "text",  # for onbox
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -