Skip to content

medcat.components.addons.relation_extraction.pad_seq

Classes:

Pad_Sequence 

Pad_Sequence(seq_pad_value: int, label_pad_value: int = -1)

Used in rel_cat.py in RelCAT to create DataLoaders for train/test datasets. collate_fn for dataloader to collate sequences of different input_ids, ent1/ent2, and label lengths into a fixed length batch. This is applied per batch and not on the whole DataLoader data, padded x sequence, y sequence, x lengths and y lengths of batch.

Parameters:

  • seq_pad_value

    (int) –

    pad value for input_ids.

  • label_pad_value

    (int, default: -1 ) –

    pad value for labels. Defaults to -1.

Attributes:

Source code in medcat-v2/medcat/components/addons/relation_extraction/pad_seq.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
def __init__(self, seq_pad_value: int, label_pad_value: int = -1):
    """Used in rel_cat.py in RelCAT to create DataLoaders for
    train/test datasets.
    collate_fn for dataloader to collate sequences of different input_ids,
    ent1/ent2, and label lengths into a fixed length batch.
    This is applied per batch and not on the whole DataLoader data,
    padded x sequence, y sequence, x lengths and y lengths of batch.

    Args:
        seq_pad_value (int): pad value for input_ids.
        label_pad_value (int): pad value for labels. Defaults to -1.
    """
    self.seq_pad_value: int = seq_pad_value
    self.label_pad_value: int = label_pad_value

label_pad_value instance-attribute 

label_pad_value: int = label_pad_value

seq_pad_value instance-attribute 

seq_pad_value: int = seq_pad_value